def test_checkPrime3(self):
# Check that providing a string input produces an error
with self.assertRaises(TypeError):
is_prime('1')
def test_checkPrime(self):
self.assertEqual(is_prime(3), True)
def test_negative_number(self):
for index in range(-1, -10, -1):
self.assertFalse(is_prime(index))
def test_is_zero_not_prime(self):
self.assertFalse(is_prime(0)) # Check if 0 returns False
def test_is_prime(self):
self.assertTrue(is_prime(5)) # Check if the function returns True
self.assertFalse(is_prime(4)) # Check if the function returns False
def test_prime(self):
self.assertTrue(is_prime(sys.argv[1]))