Fix EIP4844 create_genesis_state

ethereum · Jul 15, 2022 · 90edfb9 · 90edfb9
1 parent 28609cb
commit 90edfb9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/core/pyspec/eth2spec/test/helpers/genesis.py b/tests/core/pyspec/eth2spec/test/helpers/genesis.py
@@ -1,5 +1,5 @@
 from eth2spec.test.helpers.constants import (
-    ALTAIR, BELLATRIX, CAPELLA,
+    ALTAIR, BELLATRIX, CAPELLA, EIP4844,
     FORKS_BEFORE_ALTAIR, FORKS_BEFORE_BELLATRIX,
 )
 from eth2spec.test.helpers.keys import pubkeys
@@ -57,6 +57,9 @@ def create_genesis_state(spec, validator_balances, activation_threshold):
     elif spec.fork == BELLATRIX:
         previous_version = spec.config.ALTAIR_FORK_VERSION
         current_version = spec.config.BELLATRIX_FORK_VERSION
+    elif spec.fork == EIP4844:
+        previous_version = spec.config.BELLATRIX_FORK_VERSION
+        current_version = spec.config.EIP4844_FORK_VERSION
 
     state = spec.BeaconState(
         genesis_time=0,