Export pl.exclude() (#1508)

pola-rs · Oct 9, 2021 · acd2f8b · acd2f8b
1 parent 10c2aab
commit acd2f8b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/py-polars/polars/lazy/functions.py b/py-polars/polars/lazy/functions.py
@@ -57,6 +57,7 @@
     "concat_str",
     "concat_list",
     "collect_all",
+    "exclude",
 ]
 
 

diff --git a/py-polars/tests/test_lazy.py b/py-polars/tests/test_lazy.py
@@ -356,6 +356,12 @@ def test_regex_selection():
     assert df.select([col("^foo.*$")]).columns == ["foo", "fooey", "foobar"]
 
 
+def test_exclude_selection():
+    df = pl.DataFrame({"a": [1], "b": [1], "c": [1]}).lazy()
+
+    assert df.select([pl.exclude("a")]).columns == ["b", "c"]
+
+
 def test_literal_projection():
     df = pl.DataFrame({"a": [1, 2]})
     assert df.select([True]).dtypes == [pl.Boolean]