feat(tests): Example EVM code type parametrized test

ethereum · Jun 18, 2024 · 37a9edc · 37a9edc
1 parent 89f66a4
commit 37a9edc
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/tests/frontier/opcodes/test_dup.py b/tests/frontier/opcodes/test_dup.py
@@ -5,10 +5,11 @@
 """
 import pytest
 
-from ethereum_test_forks import Frontier, Homestead
+from ethereum_test_forks import EVMCodeType, Frontier, Homestead
 from ethereum_test_tools import Account, Alloc, Environment
 from ethereum_test_tools import Opcodes as Op
 from ethereum_test_tools import StateTestFiller, Storage, Transaction
+from ethereum_test_tools.eof.v1 import Container
 
 
 @pytest.mark.parametrize(
@@ -33,8 +34,10 @@
     ],
     ids=lambda op: str(op),
 )
+@pytest.mark.with_all_evm_code_types
 def test_dup(
     state_test: StateTestFiller,
+    evm_code_type: EVMCodeType,
     fork: str,
     dup_opcode: Op,
     pre: Alloc,
@@ -58,9 +61,11 @@ def test_dup(
     account_code += dup_opcode
 
     # Save each stack value into different keys in storage
-    account_code += sum(Op.PUSH1(i) + Op.SSTORE for i in range(0x11))
+    account_code += sum(Op.PUSH1(i) + Op.SSTORE for i in range(0x11)) + Op.STOP
 
-    account = pre.deploy_contract(account_code)
+    account = pre.deploy_contract(
+        account_code if evm_code_type == EVMCodeType.LEGACY else Container.Code(account_code)
+    )
 
     tx = Transaction(
         ty=0x0,