P71_PythonUnittest.py

# Author: OMKAR PATHAK

# This module helps to build the testcases for a particular program to test its integrity and overall execution

import unittest

def checkPrime(number):
    '''This function checks if the number is a prime number'''
    if number == 2:
        return True
    if number > 2:
        for i in range(2, number):
            if number % i == 0:
                return False
                break
            else:
                return True
                break
    else:
        return False

# Class for providing test cases
class CheckPrime(unittest.TestCase):

    def test_checkPrime(self):
        self.assertEqual(checkPrime(3), True)   # Check if the function returns the value specified in the second argument

    def test_checkPrime2(self):
        self.assertTrue(checkPrime(5))          # Check if the function returns True
        self.assertFalse(checkPrime(4))         # Check if the function returns False

    def test_checkPrime3(self):
        # Check that providing a string input produces an error
        with self.assertRaises(TypeError):
            checkPrime('1')

if __name__ == '__main__':
    unittest.main()

    # OUTPUT:
    # ----------------------------------------------------------------------
    # Ran 3 tests in 0.000s
    #  
    # OK