def test_primes(): with open('tests/data_primes.json') as file: data = json.load(file) p = iter(Primes()) for prime in data: assert next(p) == prime
def test_primes(self): with open('tests/data_primes.json') as file: data = json.load(file) p = iter(Primes()) for prime in data: with self.subTest(prime = prime): self.assertEqual(next(p), prime)
def test_primes_is_iterator(): assert '__init__' in dir(Primes) iterator = iter(Primes()) assert '__next__' in dir(iterator)
def test_is_iterator(self): self.assertIn('__init__', dir(Primes)) iterator = iter(Primes()) self.assertIn('__next__', dir(iterator))