fix some tests

vyperlang · Nov 13, 2023 · 519ec25 · 519ec25
1 parent 78de745
commit 519ec25
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/tests/functional/codegen/types/numbers/test_constants.py b/tests/functional/codegen/types/numbers/test_constants.py
@@ -4,7 +4,7 @@
 import pytest
 
 from vyper.compiler import compile_code
-from vyper.exceptions import InvalidType
+from vyper.exceptions import TypeMismatch
 from vyper.utils import MemoryPositions
 
 
@@ -151,7 +151,7 @@ def test_custom_constants_fail(get_contract, assert_compile_failed, storage_type
 def foo() -> {return_type}:
     return MY_CONSTANT
     """
-    assert_compile_failed(lambda: get_contract(code), InvalidType)
+    assert_compile_failed(lambda: get_contract(code), TypeMismatch)
 
 
 def test_constant_address(get_contract):

diff --git a/tests/functional/syntax/test_bool.py b/tests/functional/syntax/test_bool.py
@@ -37,7 +37,7 @@ def foo():
 def foo() -> bool:
     return (1 == 2) <= (1 == 1)
     """,
-        TypeMismatch,
+        InvalidOperation,
     ),
     """
 @external

diff --git a/tests/functional/syntax/test_ternary.py b/tests/functional/syntax/test_ternary.py
@@ -82,7 +82,7 @@ def foo() -> uint256:
 def foo() -> uint256:
     return 1 if TEST else 2
     """,
-        InvalidType,
+        TypeMismatch,
     ),
     (  # bad test type: variable
         """