EIP4844: Make extra sure we didn't win the jackpot

GAEAlimited · Jan 17, 2023 · c6453e2 · c6453e2
1 parent 6e5df21
commit c6453e2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/...pec/eth2spec/test/eip4844/unittests/polynomial_commitments/test_polynomial_commitments.py b/...pec/eth2spec/test/eip4844/unittests/polynomial_commitments/test_polynomial_commitments.py
@@ -42,8 +42,10 @@ def test_barycentric_outside_domain(spec, state):
     n_samples = 12
 
     for i in range(n_samples):
-        # Get a random evaluation point
+        # Get a random evaluation point and make sure it's not a root of unity
         z = rng.randint(0, spec.BLS_MODULUS - 1)
+        while z in roots_of_unity_brp:
+            z = rng.randint(0, spec.BLS_MODULUS - 1)
 
         # Get p(z) by evaluating poly in coefficient form
         p_z_coeff = eval_poly_in_coeff_form(spec, poly_coeff, z)