import sum

def test_add():
	assert sum.add(3,4)==7

def test_square():
    assert sum.square(2)==4

def test_divide():
    assert sum.divide(10,2)==5

def test_divide2():
    assert sum.divide(10,5)==2