From 6f05782121155874a0f21b6a586d4c1198a27dd7 Mon Sep 17 00:00:00 2001 From: coastalwhite Date: Tue, 10 Sep 2024 13:02:50 +0200 Subject: [PATCH] add sum and product to test --- py-polars/tests/unit/lazyframe/test_optimizations.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/py-polars/tests/unit/lazyframe/test_optimizations.py b/py-polars/tests/unit/lazyframe/test_optimizations.py index f646aedc9266..a4adf8dc435d 100644 --- 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)