Enable EIP7594 merkle proof testgen

ethereum · Apr 27, 2024 · fbccef3 · fbccef3
1 parent 3c96d41
commit fbccef3
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tests/generators/merkle_proof/main.py b/tests/generators/merkle_proof/main.py
@@ -1,14 +1,19 @@
-from eth2spec.test.helpers.constants import DENEB
-from eth2spec.gen_helpers.gen_from_tests.gen import run_state_test_generators
+from eth2spec.test.helpers.constants import DENEB, EIP7594
+from eth2spec.gen_helpers.gen_from_tests.gen import run_state_test_generators, combine_mods
 
 
 if __name__ == "__main__":
     deneb_mods = {key: 'eth2spec.test.deneb.merkle_proof.test_' + key for key in [
         'single_merkle_proof',
     ]}
+    _new_eip7594_mods = {key: 'eth2spec.test.eip7594.merkle_proof.test_' + key for key in [
+        'single_merkle_proof',
+    ]}
+    eip_7594_mods = combine_mods(_new_eip7594_mods, deneb_mods)
 
     all_mods = {
         DENEB: deneb_mods,
+        EIP7594: eip_7594_mods,
     }
 
     run_state_test_generators(runner_name="merkle_proof", all_mods=all_mods)