update MRE test

pola-rs · Dec 20, 2024 · c7808d3 · c7808d3
1 parent 08d86af
commit c7808d3
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/py-polars/tests/unit/io/test_parquet.py b/py-polars/tests/unit/io/test_parquet.py
@@ -2716,16 +2716,18 @@ def test_allow_missing_columns_predicate_pushdown_20361() -> None:
     f1 = io.BytesIO()
     f2 = io.BytesIO()
 
-    pl.DataFrame({"a": [1], "b": [1]}).write_parquet(f1)
-    pl.DataFrame({"a": [1]}).write_parquet(f2)
+    pl.DataFrame({"a": [1, 2], "b": [1, 3], "c": [False, False]}).write_parquet(f1)
+    pl.DataFrame({"a": [4, 5], "c": [True, False]}).write_parquet(f2)
 
     f1.seek(0)
     f2.seek(0)
 
     df = (
         pl.scan_parquet([f1, f2], allow_missing_columns=True)  # type: ignore[arg-type]
-        .filter(pl.col.a == pl.col.b)
+        .filter((pl.col.a == pl.col.b) | pl.col.c)
         .collect()
     )
 
-    assert_frame_equal(df, pl.DataFrame({"a": [1, 1], "b": [1, None]}))
+    assert_frame_equal(
+        df, pl.DataFrame({"a": [1, 4], "b": [1, None], "c": [False, True]})
+    )