def test_get_square_root(self): number = 4 epsilon = 0.01 root_type = 2 square_root = get_root(root_type, number, epsilon) self.assertTrue(abs(number - square_root ** root_type) < epsilon)
def test_get_cube_root_for_negative_number(self): number = -8 epsilon = 0.01 root_type = 3 square_root = get_root(root_type, number, epsilon) self.assertTrue(abs(number - square_root ** root_type) < epsilon)