From 9ffbad1b2f6924b3f34143ca4a95f1918b83e655 Mon Sep 17 00:00:00 2001 From: danielsanchezq Date: Mon, 10 Jun 2024 14:49:40 +0200 Subject: [PATCH] Expand test to different sizes --- da/kzg_rs/test_fft.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/da/kzg_rs/test_fft.py b/da/kzg_rs/test_fft.py index 15a5347..920d0a8 100644 --- a/da/kzg_rs/test_fft.py +++ b/da/kzg_rs/test_fft.py @@ -6,8 +6,9 @@ from fft import fft, ifft class TestFFT(TestCase): def test_fft_ifft(self): - roots_of_unity = [pow(23674694431658770659612952115660802947967373701506253797663184111817857449850, i, BLS_MODULUS) for i in range(1024)] - vals = list(x for x in range(1024)) - vals_fft = fft(vals, roots_of_unity, BLS_MODULUS) - self.assertEqual(vals, ifft(vals_fft, roots_of_unity, BLS_MODULUS)) + for size in [256, 512, 1024, 2048, 4096]: + roots_of_unity = [pow(23674694431658770659612952115660802947967373701506253797663184111817857449850, i, BLS_MODULUS) for i in range(size)] + vals = list(x for x in range(size)) + vals_fft = fft(vals, roots_of_unity, BLS_MODULUS) + self.assertEqual(vals, ifft(vals_fft, roots_of_unity, BLS_MODULUS))