Test that non-indexing dimensions are broadcast in both directions
(self)
| 72 | assert_equal(actual.shape, ai.shape) |
| 73 | |
| 74 | def test_broadcast(self): |
| 75 | """ Test that non-indexing dimensions are broadcast in both directions """ |
| 76 | a = np.ones((3, 4, 1)) |
| 77 | ai = np.ones((1, 2, 5), dtype=np.intp) |
| 78 | actual = take_along_axis(a, ai, axis=1) |
| 79 | assert_equal(actual.shape, (3, 2, 5)) |
| 80 | |
| 81 | |
| 82 | class TestPutAlongAxis: |
nothing calls this directly
no test coverage detected