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