From 519ec25050e7c604e369b43ec8772266d319352b Mon Sep 17 00:00:00 2001
From: tserg <8017125+tserg@users.noreply.github.com>
Date: Mon, 13 Nov 2023 22:19:34 +0800
Subject: [PATCH] fix some tests

---
 tests/functional/codegen/types/numbers/test_constants.py | 4 ++--
 tests/functional/syntax/test_bool.py                     | 2 +-
 tests/functional/syntax/test_ternary.py                  | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/tests/functional/codegen/types/numbers/test_constants.py b/tests/functional/codegen/types/numbers/test_constants.py
index 25617651ec..5f1a40d540 100644
--- 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
index 48ed37321a..5388a92b95 100644
--- 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
index 325be3e43b..3573648368 100644
--- 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
         """