Fix tests

py-polars/tests/unit/datatypes/test_temporal.py

@@ -434,7 +434,9 @@ def test_to_list() -> None:
 
 def test_rows() -> None:
     s0 = pl.Series("date", [123543, 283478, 1243]).cast(pl.Date)
-    with pytest.deprecated_call(match="`with_time_unit` is deprecated"):
+    with pytest.deprecated_call(
+        match="`ExprDateTimeNameSpace.with_time_unit` is deprecated"
+    ):
         s1 = (
             pl.Series("datetime", [a * 1_000_000 for a in [123543, 283478, 1243]])
             .cast(pl.Datetime)

py-polars/tests/unit/operations/test_drop.py

@@ -125,15 +125,3 @@ def test_drop_without_parameters() -> None:
     df = pl.DataFrame({"a": [1, 2]})
     assert_frame_equal(df.drop(), df)
     assert_frame_equal(df.lazy().drop(*[]), df.lazy())
-
-
-def test_drop_keyword_deprecated() -> None:
-    df = pl.DataFrame({"a": [1, 2], "b": [3, 4]})
-    expected = df.select("b")
-    with pytest.deprecated_call():
-        result_df = df.drop(columns="a")
-    assert_frame_equal(result_df, expected)
-
-    with pytest.deprecated_call():
-        result_lf = df.lazy().drop(columns="a")
-    assert_frame_equal(result_lf, expected.lazy())

py-polars/tests/unit/operations/test_group_by.py

@@ -873,18 +873,6 @@ def test_group_by_named() -> None:
     assert_frame_equal(result, expected)
 
 
-def test_group_by_deprecated_by_arg() -> None:
-    df = pl.DataFrame({"a": [1, 1, 2, 2, 3, 3], "b": range(6)})
-    with pytest.deprecated_call():
-        result = df.group_by(by=(pl.col("a") * 2), maintain_order=True).agg(
-            pl.col("b").min()
-        )
-    expected = df.group_by((pl.col("a") * 2), maintain_order=True).agg(
-        pl.col("b").min()
-    )
-    assert_frame_equal(result, expected)
-
-
 def test_group_by_with_null() -> None:
     df = pl.DataFrame(
         {"a": [None, None, None, None], "b": [1, 1, 2, 2], "c": ["x", "y", "z", "u"]}