feat(python): Do not import polars.testing by default (#5284)

py-polars/polars/__init__.py

@@ -14,7 +14,6 @@ def version() -> str:
     # this is only useful for documentation
     warnings.warn("polars binary missing!")
 
-from polars import testing
 from polars.cfg import Config
 from polars.convert import (
     from_arrow,
@@ -287,7 +286,6 @@ def version() -> str:
     "from_arrow",
     "from_pandas",
     # testing
-    "testing",
     "threadpool_size",
     # version
     "show_versions",

py-polars/tests/io/test_excel.py

@@ -1,6 +1,7 @@
 from pathlib import Path
 
 import polars as pl
+from polars.testing import assert_frame_equal
 
 
 def test_read_excel() -> None:
@@ -9,7 +10,7 @@ def test_read_excel() -> None:
 
     expected = pl.DataFrame({"hello": ["Row 1", "Row 2"]})
 
-    pl.testing.assert_frame_equal(df, expected)
+    assert_frame_equal(df, expected)
 
 
 def test_read_excel_all_sheets() -> None:
@@ -19,5 +20,5 @@ def test_read_excel_all_sheets() -> None:
     expected1 = pl.DataFrame({"hello": ["Row 1", "Row 2"]})
     expected2 = pl.DataFrame({"world": ["Row 3", "Row 4"]})
 
-    pl.testing.assert_frame_equal(df["Sheet1"], expected1)
-    pl.testing.assert_frame_equal(df["Sheet2"], expected2)
+    assert_frame_equal(df["Sheet1"], expected1)
+    assert_frame_equal(df["Sheet2"], expected2)

py-polars/tests/unit/test_df.py

@@ -1045,7 +1045,7 @@ def test_describe() -> None:
             "f": ["3", "0", None, None, "2020-01-01", "2022-01-01", None],
         }
     )
-    pl.testing.assert_frame_equal(df.describe(), expected)
+    assert_frame_equal(df.describe(), expected)
 
 
 def test_duration_arithmetic() -> None:

py-polars/tests/unit/test_explode.py

@@ -3,6 +3,7 @@
 import pyarrow as pa
 
 import polars as pl
+from polars.testing import assert_frame_equal
 
 
 def test_explode_empty_df_3402() -> None:
@@ -48,7 +49,7 @@ def test_explode_empty_list_4003() -> None:
 def test_explode_empty_list_4107() -> None:
     df = pl.DataFrame({"b": [[1], [2], []] * 2}).with_row_count()
 
-    pl.testing.assert_frame_equal(
+    assert_frame_equal(
         df.explode(["b"]), df.explode(["b"]).drop("row_nr").with_row_count()
     )
 

py-polars/tests/unit/test_groupby.py

@@ -5,6 +5,7 @@
 import pytest
 
 import polars as pl
+from polars.testing import assert_series_equal
 
 
 def test_groupby_sorted_empty_dataframe_3680() -> None:
@@ -143,7 +144,7 @@ def test_argsort_sort_by_groups_update__4360() -> None:
         ]
     )
 
-    pl.testing.assert_series_equal(out["result_a"], out["result_b"], check_names=False)
+    assert_series_equal(out["result_a"], out["result_b"], check_names=False)
     assert out["result_a"].to_list() == [1, 2, 3, 3, 2, 1, 2, 3, 1]
 
 

py-polars/tests/unit/test_rolling.py

@@ -7,6 +7,7 @@
 import pytest
 
 import polars as pl
+from polars.testing import assert_frame_equal
 
 if TYPE_CHECKING:
     from polars.internals.type_aliases import ClosedWindow
@@ -55,7 +56,7 @@ def test_rolling_kernels_and_groupby_rolling() -> None:
                     pl.col("values").std().alias("std"),
                 ]
             )
-            pl.testing.assert_frame_equal(out1, out2)
+            assert_frame_equal(out1, out2)
 
 
 def test_rolling_skew() -> None:

py-polars/tests/unit/test_serde.py

@@ -4,6 +4,7 @@
 from datetime import datetime, timedelta
 
 import polars as pl
+from polars.testing import assert_series_equal
 
 
 def test_pickling_simple_expression() -> None:
@@ -51,7 +52,7 @@ def test_serde_duration() -> None:
     df = df.with_columns([pl.col("a").diff(n=1).alias("a_td")])
     serde_df = pickle.loads(pickle.dumps(df))
     assert serde_df["a_td"].dtype == pl.Duration("ns")
-    pl.testing.assert_series_equal(
+    assert_series_equal(
         serde_df["a_td"],
         pl.Series("a_td", [None, timedelta(days=1)], dtype=pl.Duration("ns")),
     )

py-polars/tests/unit/test_series.py

@@ -54,7 +54,7 @@ def test_init_inputs(monkeypatch: Any) -> None:
         assert pl.Series(dtype_if_empty=pl.Utf8).dtype == pl.Utf8
         assert pl.Series([], dtype_if_empty=pl.UInt16).dtype == pl.UInt16
         # "== []" will be cast to empty Series with Utf8 dtype.
-        pl.testing.assert_series_equal(
+        assert_series_equal(
             pl.Series([], dtype_if_empty=pl.Utf8) == [], pl.Series("", dtype=pl.Boolean)
         )
         assert pl.Series(values=[True, False]).dtype == pl.Boolean