From fd1f92e12dc289ec5da06672309767252b25aae1 Mon Sep 17 00:00:00 2001 From: danielsanchezq Date: Wed, 12 Jun 2024 12:10:18 +0200 Subject: [PATCH] Fix test --- da/kzg_rs/test_fk20.py | 5 ++--- da/kzg_rs/utils.py | 3 +-- 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/da/kzg_rs/test_fk20.py b/da/kzg_rs/test_fk20.py index 2294315..c7e095f 100644 --- a/da/kzg_rs/test_fk20.py +++ b/da/kzg_rs/test_fk20.py @@ -18,12 +18,11 @@ class TestFK20(TestCase): ) def test_fk20(self): - for size in [16, 32, 64, 128]: - roots_of_unity = compute_roots_of_unity(PRIMITIVE_ROOT, size*2, BLS_MODULUS) + for size in [16, 32, 64, 128, 256]: + roots_of_unity = compute_roots_of_unity(PRIMITIVE_ROOT, size, BLS_MODULUS) rand_bytes = self.rand_bytes(size) polynomial = bytes_to_polynomial(rand_bytes) proofs = [generate_element_proof(i, polynomial, GLOBAL_PARAMETERS, roots_of_unity) for i in range(size)] fk20_proofs = fk20_generate_proofs(polynomial, GLOBAL_PARAMETERS) self.assertEqual(len(proofs), len(fk20_proofs)) self.assertEqual(proofs, fk20_proofs) - diff --git a/da/kzg_rs/utils.py b/da/kzg_rs/utils.py index 190ac22..b519f82 100644 --- a/da/kzg_rs/utils.py +++ b/da/kzg_rs/utils.py @@ -1,5 +1,4 @@ - -POWERS_OF_2 = {2**i for i in range(1, 8)} +POWERS_OF_2 = {2**i for i in range(1, 32)} def is_power_of_two(n) -> bool: