class TestPrimer2(ut.TestCase): def setUp(self): self.primer = Primer(7) def test_generate_0(self): expected = list(self.primer.generate(0)) self.assertEqual(expected, []) def test_generate_num_within_limit(self): expected = list(self.primer.generate(2)) self.assertEqual(expected, [2, 3]) @ut.skip('not now') def test_generate_num_exact_limit(self): expected = list(self.primer.generate(4)) self.assertEqual(expected, [2, 3, 5, 7]) def test_generate_num_exceed_limit(self): expected = list(self.primer.generate(5)) self.assertEqual(expected, [2, 3, 5, 7])
from primer import Primer if __name__ == '__main__': prime = Primer(20) for p in prime.generate(9): print(p)