(mgr, axis, indexer)
| 954 | @pytest.mark.parametrize("mgr", MANAGERS) |
| 955 | def test_take(self, mgr): |
| 956 | def assert_take_ok(mgr, axis, indexer): |
| 957 | mat = _as_array(mgr) |
| 958 | taken = mgr.take(indexer, axis) |
| 959 | tm.assert_numpy_array_equal( |
| 960 | np.take(mat, indexer, axis), _as_array(taken), check_dtype=False |
| 961 | ) |
| 962 | tm.assert_index_equal(mgr.axes[axis].take(indexer), taken.axes[axis]) |
| 963 | |
| 964 | for ax in range(mgr.ndim): |
| 965 | # take/fancy indexer |