diff --git a/tests/generators/epoch_processing/main.py b/tests/generators/epoch_processing/main.py index 93e9a366c..3796f76a9 100644 --- a/tests/generators/epoch_processing/main.py +++ b/tests/generators/epoch_processing/main.py @@ -37,7 +37,12 @@ if __name__ == "__main__": ]} deneb_mods = combine_mods(_new_deneb_mods, capella_mods) - electra_mods = deneb_mods + _new_electra_mods = {key: 'eth2spec.test.electra.epoch_processing.test_process_' + key for key in [ + 'effective_balance_updates', + 'pending_balance_deposits', + 'pending_consolidations', + ]} + electra_mods = combine_mods(_new_electra_mods, deneb_mods) # TODO Custody Game testgen is disabled for now # custody_game_mods = {**{key: 'eth2spec.test.custody_game.epoch_processing.test_process_' + key for key in [ diff --git a/tests/generators/operations/main.py b/tests/generators/operations/main.py index 42d698c75..fc4a8c2bf 100644 --- a/tests/generators/operations/main.py +++ b/tests/generators/operations/main.py @@ -45,21 +45,13 @@ if __name__ == "__main__": _new_electra_mods = {key: 'eth2spec.test.electra.block_processing.test_process_' + key for key in [ 'attestation', + 'consolidation' 'deposit_receipt', - 'execution_layer_exit', + 'execution_layer_withdrawal_request', + 'voluntary_exit' ]} electra_mods = combine_mods(_new_electra_mods, deneb_mods) - # TODO Custody Game testgen is disabled for now - # _new_custody_game_mods = {key: 'eth2spec.test.custody_game.block_processing.test_process_' + key for key in [ - # 'attestation', - # 'chunk_challenge', - # 'custody_key_reveal', - # 'custody_slashing', - # 'early_derived_secret_reveal', - # ]} - # custody_game_mods = combine_mods(_new_custody_game_mods, phase0_mods) - all_mods = { PHASE0: phase_0_mods, ALTAIR: altair_mods,