def test_assert_prime_number(self): ''' test the implementation of my prime number assertion function ''' self.assertEqual(assert_prime_number(0), False) self.assertEqual(assert_prime_number(1), False) self.assertEqual(assert_prime_number(4), False) self.assertEqual(assert_prime_number(7), True) self.assertEqual(assert_prime_number(571), True)
def quadratic_prime(a,b,n): ''' return the value of n^2 + an + b if prime and False otherwise ''' c = n**2 + a*n + b if assert_prime_number(c): return c else: return False