Python apply_over_axes Examples

Example #1

File: test_extras.py Project: phonx/FlipbookApp

 def test_basic(self):
     a = arange(24).reshape(2, 3, 4)
     test = apply_over_axes(np.sum, a, [0, 2])
     ctrl = np.array([[[60], [92], [124]]])
     assert_equal(test, ctrl)
     a[(a % 2).astype(np.bool)] = masked
     test = apply_over_axes(np.sum, a, [0, 2])
     ctrl = np.array([[[30], [44], [60]]])

Example #2

File: test_extras.py Project: Alanchi/numpy

 def test_basic(self):
     a = arange(24).reshape(2, 3, 4)
     test = apply_over_axes(np.sum, a, [0, 2])
     ctrl = np.array([[[60], [92], [124]]])
     assert_equal(test, ctrl)
     a[(a % 2).astype(np.bool)] = masked
     test = apply_over_axes(np.sum, a, [0, 2])
     ctrl = np.array([[[30], [44], [60]]])