From 988412ec2c3f39b5c014941905bf316dd3541267 Mon Sep 17 00:00:00 2001 From: Kevaundray Wedderburn Date: Fri, 19 Apr 2024 12:53:47 +0100 Subject: [PATCH] modify test --- .../test_polynomial_commitments.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/core/pyspec/eth2spec/test/eip7594/unittests/polynomial_commitments/test_polynomial_commitments.py b/tests/core/pyspec/eth2spec/test/eip7594/unittests/polynomial_commitments/test_polynomial_commitments.py index b6232eaeb..f7cf65d2d 100644 --- a/tests/core/pyspec/eth2spec/test/eip7594/unittests/polynomial_commitments/test_polynomial_commitments.py +++ b/tests/core/pyspec/eth2spec/test/eip7594/unittests/polynomial_commitments/test_polynomial_commitments.py @@ -67,7 +67,7 @@ def test_verify_cell_proof_batch(spec): @with_eip7594_and_later @spec_test @single_phase -def test_recover_polynomial(spec): +def test_recover_all_cells(spec): rng = random.Random(5566) # Number of samples we will be recovering from @@ -93,15 +93,15 @@ def test_recover_polynomial(spec): # Now the cells themselves known_cells_bytes = [cells_bytes[cell_id] for cell_id in cell_ids] - # Recover the data - recovered_data = spec.recover_polynomial(cell_ids, known_cells_bytes) + # Recover all of the cells + recovered_cells = spec.recover_all_cells(cell_ids, known_cells_bytes) + recovered_data = [x for xs in recovered_cells for x in xs] # Check that the original data match the non-extended portion of the recovered data assert original_polynomial == recovered_data[:len(recovered_data) // 2] - # Now flatten the cells and check that they match the entirety of the recovered data - flattened_cells = [x for xs in cells for x in xs] - assert flattened_cells == recovered_data + # Check that the recovered cells match the original cells + assert cells == recovered_data @with_eip7594_and_later