test_unit_02.py

import unittest
import math

def function_square(a):
    return a*a

def function_square_root(a):
    return math.sqrt(a)

def function_sum(a,b):
    return a+a

 
class Test_function1(unittest.TestCase):
    def test_function_square(self):
        self.assertEquals(function_square(3),4)

class Test_function2(unittest.TestCase):   
    def test_function_square_root(self):
        self.assertEquals(function_square_root(16),4)
    def test_function_square(self):
        self.assertEquals(function_sum(2,2),4)


#run the program
#python -m unittest test_unit_02.py