|
1 | 1 | from eth2spec.test.helpers.epoch_processing import ( |
2 | 2 | run_epoch_processing_with, |
3 | | - compute_state_by_epoch_processing_to, |
| 3 | + run_epoch_processing_to, |
4 | 4 | ) |
5 | 5 | from eth2spec.test.context import ( |
6 | 6 | spec_state_test, |
@@ -219,7 +219,8 @@ def test_pending_consolidation_future_epoch(spec, state): |
219 | 219 | next_epoch_with_full_participation(spec, state) |
220 | 220 |
|
221 | 221 | # Obtain state before the call to process_pending_consolidations |
222 | | - state_before_consolidation = compute_state_by_epoch_processing_to(spec, state, "process_pending_consolidations") |
| 222 | + state_before_consolidation = state.copy() |
| 223 | + run_epoch_processing_to(spec, state_before_consolidation, "process_pending_consolidations") |
223 | 224 |
|
224 | 225 | yield from run_epoch_processing_with(spec, state, "process_pending_consolidations") |
225 | 226 |
|
@@ -270,7 +271,8 @@ def test_pending_consolidation_compounding_creds(spec, state): |
270 | 271 | next_epoch_with_full_participation(spec, state) |
271 | 272 |
|
272 | 273 | # Obtain state before the call to process_pending_consolidations |
273 | | - state_before_consolidation = compute_state_by_epoch_processing_to(spec, state, "process_pending_consolidations") |
| 274 | + state_before_consolidation = state.copy() |
| 275 | + run_epoch_processing_to(spec, state_before_consolidation, "process_pending_consolidations") |
274 | 276 |
|
275 | 277 | yield from run_epoch_processing_with(spec, state, "process_pending_consolidations") |
276 | 278 |
|
@@ -325,7 +327,8 @@ def test_pending_consolidation_with_pending_deposit(spec, state): |
325 | 327 | next_epoch_with_full_participation(spec, state) |
326 | 328 |
|
327 | 329 | # Obtain state before the call to process_pending_balance_deposits |
328 | | - state_before_consolidation = compute_state_by_epoch_processing_to(spec, state, "process_pending_balance_deposits") |
| 330 | + state_before_consolidation = state.copy() |
| 331 | + run_epoch_processing_to(spec, state_before_consolidation, "process_pending_consolidations") |
329 | 332 |
|
330 | 333 | yield from run_epoch_processing_with(spec, state, "process_pending_consolidations") |
331 | 334 |
|
|
0 commit comments