| 42 | |
| 43 | class TestTake: |
| 44 | def test_1d_fill_nonna(self, dtype_fill_out_dtype): |
| 45 | dtype, fill_value, out_dtype = dtype_fill_out_dtype |
| 46 | data = np.random.default_rng(2).integers(0, 2, 4).astype(dtype) |
| 47 | indexer = [2, 1, 0, -1] |
| 48 | |
| 49 | result = algos.take_nd(data, indexer, fill_value=fill_value) |
| 50 | assert (result[[0, 1, 2]] == data[[2, 1, 0]]).all() |
| 51 | assert result[3] == fill_value |
| 52 | assert result.dtype == out_dtype |
| 53 | |
| 54 | indexer = [2, 1, 0, 1] |
| 55 | |
| 56 | result = algos.take_nd(data, indexer, fill_value=fill_value) |
| 57 | assert (result[[0, 1, 2, 3]] == data[indexer]).all() |
| 58 | assert result.dtype == dtype |
| 59 | |
| 60 | def test_2d_fill_nonna(self, dtype_fill_out_dtype): |
| 61 | dtype, fill_value, out_dtype = dtype_fill_out_dtype |