import unnecessarymath as um

def test_square():
    assert um.square(1) == 1
    assert um.square(2) == 4

def test_sqrt():
    assert um.sqrt(1.) == 1.
    assert um.sqrt(0.) == 0.
    assert um.sqrt(4.) == 2.

def power_test():
	assert um.power(3, 3) == 27
	assert um.power(2, 4) == 16
	assert um.power(5, 3) == 125