fix tests

kkrt-labs · Dec 3, 2024 · 25b8c14 · 25b8c14
1 parent 6b6d3ac
commit 25b8c14
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/cairo_zero/tests/src/kakarot/precompiles/test_precompiles.py b/cairo_zero/tests/src/kakarot/precompiles/test_precompiles.py
@@ -219,7 +219,8 @@ def test__is_precompile_rollup_precompiles(self, cairo_run, address):
             assert result == (address in ROLLUP_PRECOMPILES)
 
         @pytest.mark.parametrize(
-            "address", range(KAKAROT_PRECOMPILES[0], KAKAROT_PRECOMPILES[-1] + 2)
+            "address",
+            range(KAKAROT_PRECOMPILES[0], KAKAROT_PRECOMPILES[-1] + 2),
         )
         def test__is_precompile_kakarot_precompiles(self, cairo_run, address):
             result = cairo_run("test__is_precompile", address=address)

diff --git a/cairo_zero/tests/src/kakarot/test_state.py b/cairo_zero/tests/src/kakarot/test_state.py
@@ -5,7 +5,7 @@
     TX_ACCESS_LIST_STORAGE_KEY_COST,
 )
 
-from tests.utils.constants import ETHEREUM_PRECOMPILES, TRANSACTIONS
+from tests.utils.constants import ALL_PRECOMPILES, TRANSACTIONS
 from tests.utils.helpers import flatten_tx_access_list, merge_access_list
 from tests.utils.syscall_handler import SyscallHandler
 
@@ -90,7 +90,7 @@ def test_should_cache_precompiles(self, cairo_run):
             state = cairo_run("test__cache_precompiles")
             assert [
                 int(address, 16) for address in state["accounts"].keys()
-            ] == ETHEREUM_PRECOMPILES
+            ] == ALL_PRECOMPILES
 
         @SyscallHandler.patch("IERC20.balanceOf", lambda *_: [0, 1])
         @pytest.mark.parametrize("transaction", TRANSACTIONS)