Test that non-indexing dimensions are broadcast in both directions
(self)
| 97 | assert_equal(i_min, i_max) |
| 98 | |
| 99 | def test_broadcast(self): |
| 100 | """ Test that non-indexing dimensions are broadcast in both directions """ |
| 101 | a = np.ones((3, 4, 1)) |
| 102 | ai = np.arange(10, dtype=np.intp).reshape((1, 2, 5)) % 4 |
| 103 | put_along_axis(a, ai, 20, axis=1) |
| 104 | assert_equal(take_along_axis(a, ai, axis=1), 20) |
| 105 | |
| 106 | |
| 107 | class TestApplyAlongAxis: |
nothing calls this directly
no test coverage detected