fix(rust, python): correct ne/e_missing schema (#9257)

pola-rs · Jun 6, 2023 · 02a3a2b · 02a3a2b
1 parent ce612a9
commit 02a3a2b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/polars/polars-lazy/polars-plan/src/logical_plan/aexpr/schema.rs b/polars/polars-lazy/polars-plan/src/logical_plan/aexpr/schema.rs
@@ -64,6 +64,8 @@ impl AExpr {
                     | Operator::And
                     | Operator::LtEq
                     | Operator::GtEq
+                    | Operator::NotEqValidity
+                    | Operator::EqValidity
                     | Operator::Or => {
                         let out_field;
                         let out_name = {

diff --git a/py-polars/tests/unit/test_schema.py b/py-polars/tests/unit/test_schema.py
@@ -453,3 +453,9 @@ def test_list_null_constructor_schema() -> None:
     assert pl.Series([[]], dtype=pl.List).dtype == expected
     assert pl.DataFrame({"a": [[]]}).dtypes[0] == expected
     assert pl.DataFrame(schema={"a": pl.List}).dtypes[0] == expected
+
+
+def test_schema_ne_missing_9256() -> None:
+    df = pl.DataFrame({"a": [0, 1, None], "b": [True, False, True]})
+
+    assert df.select(pl.col("a").ne_missing(0).or_(pl.col("b")))["a"].all()