From 38a7255a5e935c3b2663613392db9d5290fc2340 Mon Sep 17 00:00:00 2001 From: adamantike Date: Mon, 28 Mar 2016 13:12:12 -0300 Subject: Set Miller-Rabin rounds based on bitsize --- tests/test_prime.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'tests') diff --git a/tests/test_prime.py b/tests/test_prime.py index 173c991..44d8d71 100644 --- a/tests/test_prime.py +++ b/tests/test_prime.py @@ -74,3 +74,17 @@ class PrimeTest(unittest.TestCase): self.assertEqual([], randints) finally: rsa.randnum.randint = orig_randint + + def test_get_primality_testing_rounds(self): + """Test round calculation for primality testing.""" + + self.assertEquals(rsa.prime.get_primality_testing_rounds(1 << 63), 10) + self.assertEquals(rsa.prime.get_primality_testing_rounds(1 << 127), 10) + self.assertEquals(rsa.prime.get_primality_testing_rounds(1 << 255), 10) + self.assertEquals(rsa.prime.get_primality_testing_rounds(1 << 511), 7) + self.assertEquals(rsa.prime.get_primality_testing_rounds(1 << 767), 7) + self.assertEquals(rsa.prime.get_primality_testing_rounds(1 << 1023), 4) + self.assertEquals(rsa.prime.get_primality_testing_rounds(1 << 1279), 4) + self.assertEquals(rsa.prime.get_primality_testing_rounds(1 << 1535), 3) + self.assertEquals(rsa.prime.get_primality_testing_rounds(1 << 2047), 3) + self.assertEquals(rsa.prime.get_primality_testing_rounds(1 << 4095), 3) -- cgit v1.2.3