Skip to content

Commit

Permalink
Revert "Temporarily skip tests due to dask/distributed#8953 (#17472)"
Browse files Browse the repository at this point in the history
This reverts commit 5190b44.
  • Loading branch information
Matt711 authored Dec 3, 2024
1 parent 12c77f3 commit f95a77e
Showing 1 changed file with 7 additions and 49 deletions.
56 changes: 7 additions & 49 deletions python/custreamz/custreamz/tests/test_dataframes.py
Original file line number Diff line number Diff line change
Expand Up @@ -216,13 +216,7 @@ def test_set_index():
assert_eq(b[0], df.set_index(df.y + 1))


def test_binary_stream_operators(request, stream):
request.applymarker(
pytest.mark.xfail(
isinstance(stream, DaskStream),
reason="https://github.com/dask/distributed/issues/8953",
)
)
def test_binary_stream_operators(stream):
df = cudf.DataFrame({"x": [1, 2, 3], "y": [4, 5, 6]})

expected = df.x + df.y
Expand All @@ -248,13 +242,7 @@ def test_index(stream):
assert_eq(L[1], df.index + 5)


def test_pair_arithmetic(request, stream):
request.applymarker(
pytest.mark.xfail(
isinstance(stream, DaskStream),
reason="https://github.com/dask/distributed/issues/8953",
)
)
def test_pair_arithmetic(stream):
df = cudf.DataFrame({"x": list(range(10)), "y": [1] * 10})

a = DataFrame(example=df.iloc[:0], stream=stream)
Expand All @@ -267,13 +255,7 @@ def test_pair_arithmetic(request, stream):
assert_eq(cudf.concat(L), (df.x + df.y) * 2)


def test_getitem(request, stream):
request.applymarker(
pytest.mark.xfail(
isinstance(stream, DaskStream),
reason="https://github.com/dask/distributed/issues/8953",
)
)
def test_getitem(stream):
df = cudf.DataFrame({"x": list(range(10)), "y": [1] * 10})

a = DataFrame(example=df.iloc[:0], stream=stream)
Expand Down Expand Up @@ -350,13 +332,7 @@ def test_repr_html(stream):
assert "1" in html


def test_setitem(request, stream):
request.applymarker(
pytest.mark.xfail(
isinstance(stream, DaskStream),
reason="https://github.com/dask/distributed/issues/8953",
)
)
def test_setitem(stream):
df = cudf.DataFrame({"x": list(range(10)), "y": [1] * 10})

sdf = DataFrame(example=df.iloc[:0], stream=stream)
Expand All @@ -380,13 +356,7 @@ def test_setitem(request, stream):
assert_eq(L[-1], df.mean())


def test_setitem_overwrites(request, stream):
request.applymarker(
pytest.mark.xfail(
isinstance(stream, DaskStream),
reason="https://github.com/dask/distributed/issues/8953",
)
)
def test_setitem_overwrites(stream):
df = cudf.DataFrame({"x": list(range(10))})
sdf = DataFrame(example=df.iloc[:0], stream=stream)
stream = sdf.stream
Expand Down Expand Up @@ -443,14 +413,8 @@ def test_setitem_overwrites(request, stream):
],
)
def test_rolling_count_aggregations(
request, op, window, m, pre_get, post_get, kwargs, stream
op, window, m, pre_get, post_get, kwargs, stream
):
request.applymarker(
pytest.mark.xfail(
isinstance(stream, DaskStream) and len(kwargs) == 0,
reason="https://github.com/dask/distributed/issues/8953",
)
)
index = pd.DatetimeIndex(
pd.date_range("2000-01-01", "2000-01-03", freq="1h")
)
Expand Down Expand Up @@ -844,13 +808,7 @@ def test_reductions_with_start_state(stream):
assert output2[0] == 360


def test_rolling_aggs_with_start_state(request, stream):
request.applymarker(
pytest.mark.xfail(
isinstance(stream, DaskStream),
reason="https://github.com/dask/distributed/issues/8953",
)
)
def test_rolling_aggs_with_start_state(stream):
example = cudf.DataFrame({"name": [], "amount": []}, dtype="float64")
sdf = DataFrame(stream, example=example)
output0 = (
Expand Down

0 comments on commit f95a77e

Please sign in to comment.