add tests

tserg · May 12, 2022 · bd1121d · bd1121d
1 parent 83d0daa
commit bd1121d
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/tests/parser/functions/test_bitwise.py b/tests/parser/functions/test_bitwise.py
@@ -24,6 +24,10 @@ def _bitwise_not(x: uint256) -> uint256:
 @external
 def _shift(x: uint256, y: int128) -> uint256:
     return shift(x, y)
+
+@external
+def _negatedShift(x: uint256, y: int128) -> uint256:
+    return shift(x, -y)
     """
 
 
@@ -55,6 +59,14 @@ def test_test_bitwise(get_contract_with_gas_estimation, evm_version):
     assert c._shift(x, -1) == x // 2
     assert c._shift(x, -3) == x // 8
     assert c._shift(x, -256) == 0
+    assert c._negatedShift(x, -3) == x * 8
+    assert c._negatedShift(x, -255) == 0
+    assert c._negatedShift(y, -255) == 2 ** 255
+    assert c._negatedShift(x, -256) == 0
+    assert c._negatedShift(x, -0) == x
+    assert c._negatedShift(x, 1) == x // 2
+    assert c._negatedShift(x, 3) == x // 8
+    assert c._negatedShift(x, 256) == 0
 
 
 @pytest.mark.parametrize("evm_version", list(EVM_VERSIONS))

diff --git a/tests/parser/syntax/test_extract32.py b/tests/parser/syntax/test_extract32.py
@@ -1,6 +1,6 @@
 import pytest
 
-from vyper.exceptions import TypeMismatch
+from vyper.exceptions import InvalidType, TypeMismatch
 
 fail_list = [
     (
@@ -28,6 +28,14 @@ def foo(inp: Bytes[32]) -> int128:
     """,
         TypeMismatch,
     ),
+    (
+        """
+@external
+def foo(inp: Bytes[32]) -> bool:
+    return extract32(inp, 0, output_type=bool)
+    """,
+        InvalidType,
+    ),
 ]