add new tests

tserg · Feb 16, 2024 · 15caa3f · 15caa3f
1 parent 161b7c7
commit 15caa3f
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 1 deletion.
diff --git a/tests/functional/codegen/types/numbers/test_decimals.py b/tests/functional/codegen/types/numbers/test_decimals.py
@@ -23,7 +23,7 @@ def test_decimal_override():
         )
 
 
-@pytest.mark.parametrize("op", ["**", "&", "|", "^"])
+@pytest.mark.parametrize("op", ["//", "**", "&", "|", "^"])
 def test_invalid_ops(get_contract, assert_compile_failed, op):
     code = f"""
 @external

diff --git a/tests/functional/codegen/types/numbers/test_signed_ints.py b/tests/functional/codegen/types/numbers/test_signed_ints.py
@@ -391,6 +391,17 @@ def foo(a: {typ}) -> {typ}:
     assert_tx_failed(lambda: c.foo(lo))
 
 
+@pytest.mark.parametrize("typ", types)
+@pytest.mark.parametrize("op", ["/"])
+def test_invalid_ops(get_contract, assert_compile_failed, typ, op):
+    code = f"""
+@external
+def foo(x: {typ}, y: {typ}) -> {typ}:
+    return x {op} y
+    """
+    assert_compile_failed(lambda: get_contract(code), InvalidOperation)
+
+
 @pytest.mark.parametrize("typ", types)
 @pytest.mark.parametrize("op", ["not"])
 def test_invalid_unary_ops(get_contract, assert_compile_failed, typ, op):

diff --git a/tests/functional/codegen/types/numbers/test_unsigned_ints.py b/tests/functional/codegen/types/numbers/test_unsigned_ints.py
@@ -216,6 +216,17 @@ def test() -> {typ}:
         assert_compile_failed(lambda: get_contract(code_template.format(typ=typ, val=val)))
 
 
+@pytest.mark.parametrize("typ", types)
+@pytest.mark.parametrize("op", ["/"])
+def test_invalid_ops(get_contract, assert_compile_failed, typ, op):
+    code = f"""
+@external
+def foo(x: {typ}, y: {typ}) -> {typ}:
+    return x {op} y
+    """
+    assert_compile_failed(lambda: get_contract(code), InvalidOperation)
+
+
 @pytest.mark.parametrize("typ", types)
 @pytest.mark.parametrize("op", ["not", "-"])
 def test_invalid_unary_ops(get_contract, assert_compile_failed, typ, op):