Fix tests

ethereum · Oct 16, 2023 · c1e33b9 · c1e33b9
1 parent 8d6b28b
commit c1e33b9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...yspec/eth2spec/test/deneb/unittests/polynomial_commitments/test_polynomial_commitments.py b/...yspec/eth2spec/test/deneb/unittests/polynomial_commitments/test_polynomial_commitments.py
@@ -113,7 +113,7 @@ def test_barycentric_outside_domain(spec):
     """
     rng = random.Random(5566)
     poly_coeff, poly_eval = get_poly_in_both_forms(spec)
-    roots_of_unity_brp = spec.bit_reversal_permutation(spec.ROOTS_OF_UNITY)
+    roots_of_unity_brp = spec.bit_reversal_permutation(spec.compute_roots_of_unity(spec.compute_roots_of_unity))
 
     assert len(poly_coeff) == len(poly_eval) == len(roots_of_unity_brp)
     n_samples = 12
@@ -147,7 +147,7 @@ def test_barycentric_within_domain(spec):
     """
     rng = random.Random(5566)
     poly_coeff, poly_eval = get_poly_in_both_forms(spec)
-    roots_of_unity_brp = spec.bit_reversal_permutation(spec.ROOTS_OF_UNITY)
+    roots_of_unity_brp = spec.bit_reversal_permutation(spec.compute_roots_of_unity(spec.compute_roots_of_unity))
 
     assert len(poly_coeff) == len(poly_eval) == len(roots_of_unity_brp)
     n = len(poly_coeff)
@@ -182,7 +182,7 @@ def test_compute_kzg_proof_within_domain(spec):
     commitment = spec.blob_to_kzg_commitment(blob)
     polynomial = spec.blob_to_polynomial(blob)
 
-    roots_of_unity_brp = spec.bit_reversal_permutation(spec.ROOTS_OF_UNITY)
+    roots_of_unity_brp = spec.bit_reversal_permutation(spec.compute_roots_of_unity(spec.compute_roots_of_unity))
 
     # Let's test some roots of unity
     for _ in range(6):