docs(python): Add "import polars.testing" to testing docstrings (#5316)…

… (#5318)
pola-rs · Oct 24, 2022 · fba2697 · fba2697
1 parent d3397fa
commit fba2697
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/py-polars/polars/testing/asserts.py b/py-polars/polars/testing/asserts.py
@@ -42,6 +42,8 @@ def assert_frame_equal(
 
     Examples
     --------
+    >>> import polars as pl
+    >>> import polars.testing
     >>> df1 = pl.DataFrame({"a": [1, 2, 3]})
     >>> df2 = pl.DataFrame({"a": [2, 3, 4]})
     >>> pl.testing.assert_frame_equal(df1, df2)  # doctest: +SKIP
@@ -119,6 +121,8 @@ def assert_series_equal(
 
     Examples
     --------
+    >>> import polars as pl
+    >>> import polars.testing
     >>> s1 = pl.Series([1, 2, 3])
     >>> s2 = pl.Series([2, 3, 4])
     >>> pl.testing.assert_series_equal(s1, s2)  # doctest: +SKIP