def testInvalidate(self): soa = SieveOfAtkin(10) soa.invalidate(2) self.assertFalse(soa.sieve[2]) soa.flip(2) self.assertTrue(soa.sieve[2])
def testFlip(self): soa = SieveOfAtkin(10) soa.flip(2) self.assertTrue(soa.sieve[2]) soa.flip(2) self.assertFalse(soa.sieve[2])
def testIsPrime(self): soa = SieveOfAtkin(100) self.assertFalse(soa.isPrime(2)) soa.flip(2) self.assertTrue(soa.isPrime(2))