Test that non-indexing dimensions are broadcast in both directions
(self)
| 84 | assert_equal(actual.shape, ai.shape) |
| 85 | |
| 86 | def test_broadcast(self): |
| 87 | """ Test that non-indexing dimensions are broadcast in both directions """ |
| 88 | a = np.ones((3, 4, 1)) |
| 89 | ai = np.ones((1, 2, 5), dtype=np.intp) |
| 90 | actual = take_along_axis(a, ai, axis=1) |
| 91 | assert_equal(actual.shape, (3, 2, 5)) |
| 92 | |
| 93 | |
| 94 | class TestPutAlongAxis: |
nothing calls this directly
no test coverage detected