fix: skip enum members during constant folding (#3235)

tests/parser/syntax/test_enum.py

@@ -137,6 +137,12 @@ def run() -> Order:
         })
     """,
     "enum Foo:\n" + "\n".join([f"    member{i}" for i in range(256)]),
+    """
+a: constant(uint256) = 1
+
+enum A:
+    a
+    """,
 ]
 
 

vyper/ast/folding.py

@@ -284,6 +284,10 @@ def replace_constant(
             if assign and node in assign.target.get_descendants(include_self=True):
                 continue
 
+        # do not replace enum members
+        if node.get_ancestor(vy_ast.EnumDef):
+            continue
+
         try:
             # note: _replace creates a copy of the replacement_node
             new_node = _replace(node, replacement_node, type_=type_)