def test_default(self): m = Mock().default_is_prime(False) m.expect_primes(toReturn=[2, 3, 5, 7, 11, 13, 17, 19]) for num in m.primes(): m.expect_is_prime(num, toReturn=True) for num in range(20): self.assertEqual(m.is_prime(num), num in m.primes())
def test_default(self): m = Mock().default_is_prime(False) m.expect_primes(toReturn=[2, 3, 5, 7, 11, 13, 17, 19]) for num in m.primes(): m.expect_is_prime(num, toReturn=True) for num in range(20): self.assertEqual(m.is_prime(num), num in m.primes())