update test

pola-rs · Oct 14, 2024 · 3947452 · 3947452
1 parent 027f115
commit 3947452
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/py-polars/tests/unit/test_selectors.py b/py-polars/tests/unit/test_selectors.py
@@ -818,7 +818,7 @@ def test_selector_list_of_lists_18499() -> None:
 
 def test_invalid_selector() -> None:
     df = pl.DataFrame(data={"x": [1, 2], "z": ["a", "b"]})
+    sel = pl.selectors.all() + pl.col("x")
+    assert isinstance(sel, pl.selectors._selector_proxy_)
     with pytest.raises(InvalidOperationError, match="invalid selector expression"):
-        df.drop(pl.col("x", "z") + 2)  # type: ignore[arg-type]
-    with pytest.raises(InvalidOperationError, match="invalid selector expression"):
-        df.drop(pl.col("x") + 2)  # type: ignore[arg-type]
+        df.drop(sel)