Python prime Examples

Example #1

File: test_generator.py Project: tbrack/SP_Online_Course2_2018

def test_prime():
    g = gen.prime()
    for val in [
            2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43, 47, 53, 59, 61,
            67, 71, 73, 79, 83, 89, 97
    ]:
        assert next(g) == val

Example #2

File: test_generators.py Project: kmsnyde/SP_Online_Course2_2018

 def test_prime(self):
     gen = g.prime()
     seq = [2, 3, 5, 7, 11, 13, 17, 19, 23]
     for s in seq:
         self.assertEqual(next(gen), s)