chore: factor drop_test out of test_common (#586)

Co-authored-by: Marco Edward Gorelli <[email protected]>

tests/frame/drop_test.py

@@ -0,0 +1,21 @@
+from __future__ import annotations
+
+from typing import Any
+
+import pytest
+
+import narwhals.stable.v1 as nw
+
+
+@pytest.mark.parametrize(
+    ("drop", "left"),
+    [
+        (["a"], ["b", "z"]),
+        (["a", "b"], ["z"]),
+    ],
+)
+def test_drop(constructor: Any, drop: list[str], left: list[str]) -> None:
+    data = {"a": [1, 3, 2], "b": [4, 4, 6], "z": [7.0, 8, 9]}
+    df = nw.from_native(constructor(data))
+    assert df.drop(drop).columns == left
+    assert df.drop(*drop).columns == left

tests/frame/test_common.py

@@ -156,19 +156,6 @@ def test_expr_na(constructor_with_lazy: Any) -> None:
     compare_dicts(result_nna, expected)
 
 
-@pytest.mark.parametrize(
-    ("drop", "left"),
-    [
-        (["a"], ["b", "z"]),
-        (["a", "b"], ["z"]),
-    ],
-)
-def test_drop(constructor: Any, drop: list[str], left: list[str]) -> None:
-    df = nw.from_native(constructor(data))
-    assert df.drop(drop).columns == left
-    assert df.drop(*drop).columns == left
-
-
 def test_lazy(constructor: Any) -> None:
     df = nw.from_native(constructor({"a": [1, 2, 3]}), eager_only=True)
     result = df.lazy()