feat(polars): add bitwise operations (bit_and, bit_or, bit_xor) (…

…#10634) Co-authored-by: Phillip Cloud <[email protected]>
ibis-project · Dec 31, 2024 · 56a66de · 56a66de
1 parent 4583dee
commit 56a66de
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/ibis/backends/polars/compiler.py b/ibis/backends/polars/compiler.py
@@ -717,6 +717,9 @@ def struct_column(op, **kw):
     ops.Median: "median",
     ops.Min: "min",
     ops.Sum: "sum",
+    ops.BitOr: "bitwise_or",
+    ops.BitAnd: "bitwise_and",
+    ops.BitXor: "bitwise_xor",
 }
 
 

diff --git a/ibis/backends/tests/test_aggregation.py b/ibis/backends/tests/test_aggregation.py
@@ -490,7 +490,7 @@ def mean_and_std(v):
             id="bit_and",
             marks=[
                 pytest.mark.notimpl(
-                    ["polars", "mssql", "exasol"],
+                    ["mssql", "exasol"],
                     raises=com.OperationNotDefinedError,
                 ),
                 pytest.mark.notimpl(["druid"], strict=False, raises=AssertionError),
@@ -505,7 +505,7 @@ def mean_and_std(v):
             id="bit_or",
             marks=[
                 pytest.mark.notimpl(
-                    ["polars", "mssql", "exasol"],
+                    ["mssql", "exasol"],
                     raises=com.OperationNotDefinedError,
                 ),
                 pytest.mark.notyet(
@@ -519,7 +519,7 @@ def mean_and_std(v):
             id="bit_xor",
             marks=[
                 pytest.mark.notimpl(
-                    ["polars", "mssql", "exasol"],
+                    ["mssql", "exasol"],
                     raises=com.OperationNotDefinedError,
                 ),
                 pytest.mark.notyet(