diff options
Diffstat (limited to 'numpy/lib/tests/test_shape_base.py')
| -rw-r--r-- | numpy/lib/tests/test_shape_base.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/numpy/lib/tests/test_shape_base.py b/numpy/lib/tests/test_shape_base.py index 0579d987d..a4e771876 100644 --- a/numpy/lib/tests/test_shape_base.py +++ b/numpy/lib/tests/test_shape_base.py @@ -17,6 +17,13 @@ class TestApplyAlongAxis(TestCase): [[27,30,33],[36,39,42],[45,48,51]]) +class TestApplyOverAxes(TestCase): + def test_simple(self): + a = arange(24).reshape(2,3,4) + aoa_a = apply_over_axes(sum, a, [0,2]) + assert_array_equal(aoa_a, array([[[60],[92],[124]]])) + + class TestArraySplit(TestCase): def test_integer_0_split(self): a = arange(10) |