add sum and product to test

pola-rs · Sep 10, 2024 · 6f05782 · 6f05782
1 parent e2e7fe9
commit 6f05782
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/py-polars/tests/unit/lazyframe/test_optimizations.py b/py-polars/tests/unit/lazyframe/test_optimizations.py
@@ -258,8 +258,8 @@ def test_collapse_joins_combinations() -> None:
 
     exprs = []
 
-    for lhs in [pl.col.a, pl.col.b, pl.col.x, pl.lit(1)]:
-        for rhs in [pl.col.a, pl.col.b, pl.col.x, pl.lit(1)]:
+    for lhs in [pl.col.a, pl.col.b, pl.col.x, pl.lit(1), pl.col.a + pl.col.b]:
+        for rhs in [pl.col.a, pl.col.b, pl.col.x, pl.lit(1), pl.col.a * pl.col.x]:
             for cmp in ["__eq__", "__ge__", "__lt__"]:
                 e = (getattr(lhs, cmp))(rhs)
                 exprs.append(e)