🐛 fix: check node.left in constant_folding

src/expr_simplifier/transforms/constant_folding.py

@@ -22,7 +22,9 @@ def visit(self, node: ast.AST) -> ast.AST:
             return fold_to_constant(node)
         if isinstance(node, ast.BoolOp) and all(isinstance(value, ast.Constant) for value in node.values):
             return fold_to_constant(node)
-        if isinstance(node, ast.Compare) and all(isinstance(comp, ast.Constant) for comp in node.comparators):
+        if isinstance(node, ast.Compare) and all(
+            isinstance(comp, ast.Constant) for comp in [node.left, *node.comparators]
+        ):
             return fold_to_constant(node)
         if isinstance(node, ast.JoinedStr) and all(
             isinstance(value, ast.Constant)

tests/test_transforms/test_constant_folding.py

@@ -19,6 +19,7 @@
         ("(1 > 2) * a", "False * a"),
         ("""f'Hello {"World"}!'""", "'Hello World!'"),
         ("(___x := 1) + 2 + ___x", "4"),
+        ("a == 1", "a == 1"),
     ],
 )
 def test_constant_folding(expr: str, expected: str):