additional test

pola-rs · Jun 9, 2024 · 675cca5 · 675cca5
1 parent da5414e
commit 675cca5
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 10 deletions.
diff --git a/py-polars/polars/selectors.py b/py-polars/polars/selectors.py
@@ -316,7 +316,7 @@ def __repr__(self) -> str:
             else:
                 str_params = ", ".join(
                     (repr(v)[1:-1] if k.startswith("*") else f"{k}={v!r}")
-                    for k, v in (params or {}).items()
+                    for k, v in params.items()
                 ).rstrip(",")
                 return f"cs.{selector_name}({str_params})"
 

diff --git a/py-polars/tests/unit/test_selectors.py b/py-polars/tests/unit/test_selectors.py
@@ -665,15 +665,19 @@ def test_selector_sets(df: pl.DataFrame) -> None:
     }
 
     # exclusive or
-    assert df.select((cs.matches("e|g")) ^ cs.numeric()).schema == OrderedDict(
-        {
-            "abc": pl.UInt16,
-            "bbb": pl.UInt32,
-            "eee": pl.Boolean,
-            "fgg": pl.Boolean,
-            "ghi": pl.Time,
-        }
-    )
+    for selected in (
+        df.select((cs.matches("e|g")) ^ cs.numeric()),
+        df.select((cs.contains("b", "g")) ^ pl.col("eee")),
+    ):
+        assert selected.schema == OrderedDict(
+            {
+                "abc": pl.UInt16,
+                "bbb": pl.UInt32,
+                "eee": pl.Boolean,
+                "fgg": pl.Boolean,
+                "ghi": pl.Time,
+            }
+        )
 
 
 def test_selector_dispatch_default_operator() -> None: