Test that non-indexing dimensions are broadcast in both directions
(self)
| 109 | assert_equal(i_min, i_max) |
| 110 | |
| 111 | def test_broadcast(self): |
| 112 | """ Test that non-indexing dimensions are broadcast in both directions """ |
| 113 | a = np.ones((3, 4, 1)) |
| 114 | ai = np.arange(10, dtype=np.intp).reshape((1, 2, 5)) % 4 |
| 115 | put_along_axis(a, ai, 20, axis=1) |
| 116 | assert_equal(take_along_axis(a, ai, axis=1), 20) |
| 117 | |
| 118 | def test_invalid(self): |
| 119 | """ Test invalid inputs """ |
nothing calls this directly
no test coverage detected