Merge pull request #3732 from ethereum/enable-eip7594-merkle-proof-te…

…stgen Enable EIP7594 merkle proof testgen
ethereum · Apr 27, 2024 · 186c943 · 186c943
2 parents 3c96d41 + b681552
commit 186c943
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/tests/core/pyspec/eth2spec/gen_helpers/gen_from_tests/gen.py b/tests/core/pyspec/eth2spec/gen_helpers/gen_from_tests/gen.py
@@ -77,7 +77,7 @@ def get_provider(create_provider_fn: Callable[[SpecForkName, PresetBaseName, str
 
 def get_create_provider_fn(runner_name: str) -> Callable[[SpecForkName, str, str, PresetBaseName], TestProvider]:
     def prepare_fn() -> None:
-        bls.use_milagro()
+        bls.use_fastest()
         return
 
     def create_provider(fork_name: SpecForkName, preset_name: PresetBaseName,

diff --git a/tests/core/pyspec/eth2spec/test/eip7594/merkle_proof/test_single_merkle_proof.py b/tests/core/pyspec/eth2spec/test/eip7594/merkle_proof/test_single_merkle_proof.py
@@ -22,7 +22,7 @@
 
 
 def _run_blob_kzg_commitments_merkle_proof_test(spec, state, rng=None):
-    opaque_tx, blobs, blob_kzg_commitments, proofs = get_sample_opaque_tx(spec, blob_count=1)
+    opaque_tx, blobs, blob_kzg_commitments, _ = get_sample_opaque_tx(spec, blob_count=1)
     if rng is None:
         block = build_empty_block_for_next_slot(spec, state)
     else:

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)