def test_cubes_is_iterator(): assert '__init__' in dir(Cubes) iterator = iter(Cubes()) assert '__next__' in dir(iterator)
def test_is_iterator(self): self.assertIn('__init__', dir(Cubes)) iterator = iter(Cubes()) self.assertIn('__next__', dir(iterator))
def test_cubes(): c = iter(Cubes()) for i in range(1, 1001): assert next(c) == i**3
def test_cubes(self): c = iter(Cubes()) for i in range(1, 1001): with self.subTest(i = i): self.assertEqual(next(c), i ** 3)