import my_module print my_module.nth_prime(10001)
def test_nth_prime(self): self.assertEqual(my_module.nth_prime(7), 17)