fix exception type in test_slice_bytes

vyperlang · Jul 16, 2023 · 727be2a · 727be2a
1 parent 856c2a6
commit 727be2a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/parser/functions/test_slice.py b/tests/parser/functions/test_slice.py
@@ -2,7 +2,7 @@
 import pytest
 from hypothesis import given, settings
 
-from vyper.exceptions import ArgumentException
+from vyper.exceptions import ArgumentException, TypeMismatch
 
 _fun_bytes32_bounds = [(0, 32), (3, 29), (27, 5), (0, 5), (5, 3), (30, 2)]
 
@@ -143,7 +143,7 @@ def _get_contract():
         or (literal_start and start > data_length)
         or (literal_length and length < 1)
     ):
-        assert_compile_failed(lambda: _get_contract(), ArgumentException)
+        assert_compile_failed(lambda: _get_contract(), (ArgumentException, TypeMismatch))
     elif len(bytesdata) > data_length:
         # deploy fail
         assert_tx_failed(lambda: _get_contract())