Exemplo n.º 1
0
    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())
Exemplo n.º 2
0
    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())