Added function to test if a number is prime

This commit is contained in:
2021-06-30 18:42:35 -04:00
parent 78752e3b90
commit 622cde6bfe
2 changed files with 36 additions and 2 deletions

View File

@@ -42,6 +42,29 @@ class TestAlgorithms(unittest.TestCase):
answer = Algorithms.getNumPrimes(numPrimes) #Get the answer from the function
self.assertEqual(correctAnswer, answer, "getNumPrimes failed the first test")
#This function tests the isPrime function
def testIsPrime(self):
#Test 1
num = 2
correctAnswer = True
answer = Algorithms.isPrime(num)
self.assertEqual(correctAnswer, answer, "isPrime failed the first test")
#Test 2
num = 97
correctAnswer = True
answer = Algorithms.isPrime(num)
self.assertEqual(correctAnswer, answer, "isPrime failed the second test")
#Test 3
num = 1000
correctAnswer = False
answer = Algorithms.isPrime(num)
self.assertEqual(correctAnswer, answer, "isPrime failed the third test")
#Test 4
num = 1
correctAnswer = False
answer = Algorithms.isPrime(num)
self.assertEqual(correctAnswer, answer, "isPrime failed the fourth test")
#This function tests the getFactors function
def testGetFactors(self):
#Test 1
@@ -203,9 +226,9 @@ if __name__ == "__main__":
unittest.main()
"""Results:
............
.............
----------------------------------------------------------------------
Ran 12 tests in 0.002s
Ran 13 tests in 0.002s
OK
"""