from src.mycode import * 



def test_the_square_function(): 
    assert square(4) == 16 

def test_the_cube_function(): 
    assert cube(4) == 64 

def test_the_quad_function(): 
    assert quad(4) == 256 


