fix: bitwise not constant folding (#3222)

vyperlang · Jan 20, 2023 · deb3378 · deb3378
1 parent d59ffaa
commit deb3378
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/tests/builtins/folding/test_bitwise.py b/tests/builtins/folding/test_bitwise.py
@@ -40,9 +40,9 @@ def foo(a: uint256) -> uint256:
     """
     contract = get_contract(source)
 
-    vyper_ast = vy_ast.parse_to_ast(f"bitwise_not({value})")
+    vyper_ast = vy_ast.parse_to_ast(f"~{value}")
     old_node = vyper_ast.body[0].value
-    new_node = vy_fn.BitwiseNot().evaluate(old_node)
+    new_node = old_node.evaluate()
 
     assert contract.foo(value) == new_node.value
 

diff --git a/vyper/ast/nodes.py b/vyper/ast/nodes.py
@@ -937,7 +937,9 @@ class Invert(Operator):
     __slots__ = ()
     _description = "bitwise not"
     _pretty = "~"
-    _op = operator.inv
+
+    def _op(self, value):
+        return (2 ** 256 - 1) ^ value
 
 
 class BinOp(ExprNode):