def test_is_prime(self):
primes = [2, 3, 5, 7, 13, 17, 104729, 1299709, 1300391]
not_primes = [4, 8, 16, 27, 1000, 100000, 10000000000000]
for p in primes:
self.assertTrue(is_prime(p))
for np in not_primes:
self.assertFalse(is_prime(np))
def __init__(self, a, b, p):
if not is_prime(p):
raise BadCurve
else:
self.coef_a = a
self.coef_b = b
self.modulus = p
