def test_prime_gen(self):
self.assertTrue(_is_prime(_get_prime(1024)))
def test_prime(self):
for n in [3, 5, 7, 13, 17, 31]:
self.assertTrue(_is_prime(n))
for n in [0, 1, 4, 6, 9, 15, 20, 100]:
self.assertFalse(_is_prime(n))
