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))