Skip to content

Commit f46043d

Browse files
committed
These tests now pass
1 parent ceddcb9 commit f46043d

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

python/cudf_polars/tests/test_rolling.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,10 @@
99

1010
import polars as pl
1111

12-
from cudf_polars.testing.asserts import assert_ir_translation_raises
12+
from cudf_polars.testing.asserts import (
13+
assert_gpu_result_equal,
14+
assert_ir_translation_raises,
15+
)
1316

1417

1518
@pytest.fixture
@@ -46,10 +49,10 @@ def test_datetime_rolling(df, closed, period):
4649
max_a=pl.max("values"),
4750
)
4851

49-
assert_ir_translation_raises(q, NotImplementedError)
52+
assert_gpu_result_equal(q)
5053

5154

5255
def test_calendrical_period_unsupported(df):
53-
q = df.rolling("dt", period="1m", closed="right").agg(sum=pl.sum("values"))
56+
q = df.rolling("dt", period="1mo", closed="right").agg(sum=pl.sum("values"))
5457

5558
assert_ir_translation_raises(q, NotImplementedError)

0 commit comments

Comments
 (0)