def test_2d(self):
x = [[1, 2, 3],
[4, 5, 6],
[7, 8, 9]]
assert_allclose(mean(x), [2, 5, 8])
assert_allclose(mean(x, axis=-1), [2, 5, 8])
assert_allclose(mean(x, axis=1), [2, 5, 8])
assert_allclose(mean(x, axis=0), [4, 5, 6])
def test_1d(self):
x = [1, 2, 3, 4, 5, 6]
assert mean(x) == 3.5
