chore: prepare tests for v2 (#2380)

Author: MarcoGorelli

narwhals/_pandas_like/dataframe.py

@@ -962,7 +962,7 @@ def to_numpy(self: Self, dtype: Any = None, *, copy: bool | None = None) -> _2DA
     def to_pandas(self: Self) -> pd.DataFrame:
         if self._implementation is Implementation.PANDAS:
             return self.native
-        elif self._implementation is Implementation.CUDF:  # pragma: no cover
+        elif self._implementation is Implementation.CUDF:
             return self.native.to_pandas()
         elif self._implementation is Implementation.MODIN:
             return self.native._to_pandas()

pyproject.toml

@@ -255,12 +255,15 @@ omit = [
 exclude_also = [
   "if sys.version_info() <",
   "if .*implementation is Implementation.CUDF",
+  "if .*implementation is Implementation.MODIN",
   "if .*implementation is Implementation.PYSPARK",
   "if .*implementation.is_cudf",
+  "if .*implementation.is_modin",
   "if .*implementation.is_pyspark",
   'request.applymarker\(pytest.mark.xfail',
   'backend_version <',
   'if "cudf" in str\(constructor',
+  'if "modin" in str\(constructor',
   'if "pyspark" in str\(constructor',
   'pytest.skip\('
 ]

tests/dependencies/is_narwhals_dataframe_test.py

@@ -3,7 +3,7 @@
 from typing import TYPE_CHECKING
 
 import narwhals as nw
-import narwhals.stable.v1 as nws
+import narwhals.stable.v1 as nw_v1
 from narwhals.stable.v1.dependencies import is_narwhals_dataframe
 
 if TYPE_CHECKING:
@@ -14,5 +14,5 @@ def test_is_narwhals_dataframe(constructor_eager: ConstructorEager) -> None:
     df = constructor_eager({"col1": [1, 2], "col2": [3, 4]})
 
     assert is_narwhals_dataframe(nw.from_native(df))
-    assert is_narwhals_dataframe(nws.from_native(df))
+    assert is_narwhals_dataframe(nw_v1.from_native(df))
     assert not is_narwhals_dataframe(df)

tests/dependencies/is_narwhals_lazyframe_test.py

@@ -3,7 +3,7 @@
 from typing import TYPE_CHECKING
 
 import narwhals as nw
-import narwhals.stable.v1 as nws
+import narwhals.stable.v1 as nw_v1
 from narwhals.stable.v1.dependencies import is_narwhals_lazyframe
 from tests.utils import Constructor
 
@@ -15,5 +15,5 @@ def test_is_narwhals_lazyframe(constructor: Constructor) -> None:
     lf = constructor({"a": [1, 2, 3]})
 
     assert is_narwhals_lazyframe(nw.from_native(lf).lazy())
-    assert is_narwhals_lazyframe(nws.from_native(lf).lazy())
+    assert is_narwhals_lazyframe(nw_v1.from_native(lf).lazy())
     assert not is_narwhals_lazyframe(lf)

tests/dependencies/is_narwhals_series_test.py

@@ -3,7 +3,7 @@
 from typing import TYPE_CHECKING
 
 import narwhals as nw
-import narwhals.stable.v1 as nws
+import narwhals.stable.v1 as nw_v1
 from narwhals.stable.v1.dependencies import is_narwhals_series
 
 if TYPE_CHECKING:
@@ -14,5 +14,5 @@ def test_is_narwhals_series(constructor_eager: ConstructorEager) -> None:
     df = constructor_eager({"col1": [1, 2], "col2": [3, 4]})
 
     assert is_narwhals_series(nw.from_native(df, eager_only=True)["col1"])
-    assert is_narwhals_series(nws.from_native(df, eager_only=True)["col1"])
+    assert is_narwhals_series(nw_v1.from_native(df, eager_only=True)["col1"])
     assert not is_narwhals_series(nw.from_native(df, eager_only=True)["col1"].to_native())

tests/dtypes_test.py

@@ -11,7 +11,7 @@
 import pyarrow as pa
 import pytest
 
-import narwhals.stable.v1 as nw
+import narwhals as nw
 from tests.utils import PANDAS_VERSION
 from tests.utils import POLARS_VERSION
 from tests.utils import PYARROW_VERSION
@@ -308,7 +308,13 @@ def test_dtype_is_x() -> None:
     )
 
     is_signed_integer = {nw.Int8, nw.Int16, nw.Int32, nw.Int64, nw.Int128}
-    is_unsigned_integer = {nw.UInt8, nw.UInt16, nw.UInt32, nw.UInt64, nw.UInt128}
+    is_unsigned_integer = {
+        nw.UInt8,
+        nw.UInt16,
+        nw.UInt32,
+        nw.UInt64,
+        nw.UInt128,
+    }
     is_float = {nw.Float32, nw.Float64}
     is_decimal = {nw.Decimal}
     is_temporal = {nw.Datetime, nw.Date, nw.Duration, nw.Time}
@@ -353,7 +359,8 @@ def test_huge_int_to_native() -> None:
     result = (
         nw.from_native(rel)
         .with_columns(
-            a_int=nw.col("a").cast(nw.Int128()), a_unit=nw.col("a").cast(nw.UInt128())
+            a_int=nw.col("a").cast(nw.Int128()),
+            a_unit=nw.col("a").cast(nw.UInt128()),
         )
         .select("a_int", "a_unit")
         .to_native()

tests/expr_and_series/abs_test.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-import narwhals.stable.v1 as nw
+import narwhals as nw
 from tests.utils import Constructor
 from tests.utils import ConstructorEager
 from tests.utils import assert_equal_data

tests/expr_and_series/all_horizontal_test.py

@@ -4,7 +4,7 @@
 
 import pytest
 
-import narwhals.stable.v1 as nw
+import narwhals as nw
 from tests.utils import POLARS_VERSION
 from tests.utils import Constructor
 from tests.utils import ConstructorEager

tests/expr_and_series/any_all_test.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-import narwhals.stable.v1 as nw
+import narwhals as nw
 from tests.utils import Constructor
 from tests.utils import ConstructorEager
 from tests.utils import assert_equal_data

tests/expr_and_series/any_horizontal_test.py

@@ -4,7 +4,7 @@
 
 import pytest
 
-import narwhals.stable.v1 as nw
+import narwhals as nw
 from tests.utils import Constructor
 from tests.utils import assert_equal_data