From 313ed1a962c2e8117212c3afc322d1b6bf8c2499 Mon Sep 17 00:00:00 2001
From: Hsiao-Wei Wang <hsiaowei.eth@gmail.com>
Date: Thu, 18 Apr 2024 17:15:43 +0800
Subject: [PATCH] Fix tests

---
 .../pyspec/eth2spec/test/electra/epoch_processing/__init__.py | 0
 .../test_process_effective_balance_updates.py                 | 4 ++--
 tests/generators/operations/main.py                           | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)
 create mode 100644 tests/core/pyspec/eth2spec/test/electra/epoch_processing/__init__.py

diff --git a/tests/core/pyspec/eth2spec/test/electra/epoch_processing/__init__.py b/tests/core/pyspec/eth2spec/test/electra/epoch_processing/__init__.py
new file mode 100644
index 0000000000..e69de29bb2
diff --git a/tests/core/pyspec/eth2spec/test/phase0/epoch_processing/test_process_effective_balance_updates.py b/tests/core/pyspec/eth2spec/test/phase0/epoch_processing/test_process_effective_balance_updates.py
index b13bf9cd47..3fffcb0369 100644
--- a/tests/core/pyspec/eth2spec/test/phase0/epoch_processing/test_process_effective_balance_updates.py
+++ b/tests/core/pyspec/eth2spec/test/phase0/epoch_processing/test_process_effective_balance_updates.py
@@ -9,7 +9,7 @@
 @with_all_phases
 @spec_state_test
 def test_effective_balance_hysteresis(spec, state):
-    run_test_effective_balance_hysteresis(spec, state)
+    yield from run_test_effective_balance_hysteresis(spec, state)
 
 
 def run_test_effective_balance_hysteresis(spec, state, with_compounding_credentials=False):
@@ -18,7 +18,7 @@ def run_test_effective_balance_hysteresis(spec, state, with_compounding_credenti
     # Then overwrite the balances, we only want to focus to be on the hysteresis based changes.
     run_epoch_processing_to(spec, state, 'process_effective_balance_updates')
     # Set some edge cases for balances
-    max = spec.MAX_EFFECTIVE_BALANCE_ELECTRA if with_compounding_credentials else spec.MIN_ACTIVATION_BALANCE
+    max = spec.MAX_EFFECTIVE_BALANCE_ELECTRA if with_compounding_credentials else spec.MAX_EFFECTIVE_BALANCE
     min = spec.config.EJECTION_BALANCE
     inc = spec.EFFECTIVE_BALANCE_INCREMENT
     div = spec.HYSTERESIS_QUOTIENT
diff --git a/tests/generators/operations/main.py b/tests/generators/operations/main.py
index fc4a8c2bfb..0d203fca6f 100644
--- a/tests/generators/operations/main.py
+++ b/tests/generators/operations/main.py
@@ -45,7 +45,7 @@
 
     _new_electra_mods = {key: 'eth2spec.test.electra.block_processing.test_process_' + key for key in [
         'attestation',
-        'consolidation'
+        'consolidation',
         'deposit_receipt',
         'execution_layer_withdrawal_request',
         'voluntary_exit'