Complex

Author: dflook

src/python_minifier/__init__.py

@@ -155,9 +155,9 @@ def minify(
         module = RemoveExplicitReturnNone()(module)
 
     if constant_folding:
-        print(print_ast(module))
+        #print(print_ast(module))
         module = FoldConstants()(module)
-        print(print_ast(module))
+        #print(print_ast(module))
 
     bind_names(module)
     resolve_names(module)

src/python_minifier/transforms/constant_folding.py

@@ -38,12 +38,14 @@ def visit_BinOp(self, node):
             new_node = ast.Bytes(s=value)
         elif isinstance(value, bool):
             new_node = ast.NameConstant(value=value)
-        elif isinstance(value, (int, float, complex)):
+        elif isinstance(value, (int, float)):
             if value < 0:
                 # Represent negative numbers as a USub UnaryOp, so that the ast roundtrip is correct
                 new_node = ast.UnaryOp(op=ast.USub(), operand=ast.Num(n=-value))
             else:
                 new_node = ast.Num(n=value)
+        elif isinstance(value, complex):
+            new_node = ast.Num(n=value)
         else:
             return node