test_pytest_02.py

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+b
 
def test_function_square():
    assert function_square(2) == 4
def test_function_square_root():
    assert function_square_root(16) == 4
def test_function_sum():
    assert function_sum(2,3) == 5

test_function_square()
test_function_square()
test_function_sum()


#Run the file
#pytest test_pytest_02.py