Simple example of Python unittest module

OmkarPathak · OmkarPathak · commit b6c3250642bf · 2017-07-02T07:52:49.000+05:30
diff --git a/Programs/P71_PythonUnittest.py b/Programs/P71_PythonUnittest.py
@@ -0,0 +1,44 @@
+# 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&nbsp;3&nbsp;tests&nbsp;in&nbsp;0.000s
+    # &nbsp;
+    # OK
