|
1 | 1 | import numpy as np
|
| 2 | +import pytest |
2 | 3 |
|
3 | 4 | from pandas import (
|
4 | 5 | Categorical,
|
|
12 | 13 | class TestReindex:
|
13 | 14 | def test_reindex_list_non_unique(self):
|
14 | 15 | # GH#11586
|
| 16 | + msg = "cannot reindex on an axis with duplicate labels" |
15 | 17 | ci = CategoricalIndex(["a", "b", "c", "a"])
|
16 |
| - with tm.assert_produces_warning(FutureWarning, match="non-unique"): |
17 |
| - res, indexer = ci.reindex(["a", "c"]) |
18 |
| - |
19 |
| - tm.assert_index_equal(res, Index(["a", "a", "c"]), exact=True) |
20 |
| - tm.assert_numpy_array_equal(indexer, np.array([0, 3, 2], dtype=np.intp)) |
| 18 | + with pytest.raises(ValueError, match=msg): |
| 19 | + ci.reindex(["a", "c"]) |
21 | 20 |
|
22 | 21 | def test_reindex_categorical_non_unique(self):
|
| 22 | + msg = "cannot reindex on an axis with duplicate labels" |
23 | 23 | ci = CategoricalIndex(["a", "b", "c", "a"])
|
24 |
| - with tm.assert_produces_warning(FutureWarning, match="non-unique"): |
25 |
| - res, indexer = ci.reindex(Categorical(["a", "c"])) |
26 |
| - |
27 |
| - exp = CategoricalIndex(["a", "a", "c"], categories=["a", "c"]) |
28 |
| - tm.assert_index_equal(res, exp, exact=True) |
29 |
| - tm.assert_numpy_array_equal(indexer, np.array([0, 3, 2], dtype=np.intp)) |
| 24 | + with pytest.raises(ValueError, match=msg): |
| 25 | + ci.reindex(Categorical(["a", "c"])) |
30 | 26 |
|
31 | 27 | def test_reindex_list_non_unique_unused_category(self):
|
| 28 | + msg = "cannot reindex on an axis with duplicate labels" |
32 | 29 | ci = CategoricalIndex(["a", "b", "c", "a"], categories=["a", "b", "c", "d"])
|
33 |
| - with tm.assert_produces_warning(FutureWarning, match="non-unique"): |
34 |
| - res, indexer = ci.reindex(["a", "c"]) |
35 |
| - exp = Index(["a", "a", "c"], dtype="object") |
36 |
| - tm.assert_index_equal(res, exp, exact=True) |
37 |
| - tm.assert_numpy_array_equal(indexer, np.array([0, 3, 2], dtype=np.intp)) |
| 30 | + with pytest.raises(ValueError, match=msg): |
| 31 | + ci.reindex(["a", "c"]) |
38 | 32 |
|
39 | 33 | def test_reindex_categorical_non_unique_unused_category(self):
|
| 34 | + msg = "cannot reindex on an axis with duplicate labels" |
40 | 35 | ci = CategoricalIndex(["a", "b", "c", "a"], categories=["a", "b", "c", "d"])
|
41 |
| - with tm.assert_produces_warning(FutureWarning, match="non-unique"): |
42 |
| - res, indexer = ci.reindex(Categorical(["a", "c"])) |
43 |
| - exp = CategoricalIndex(["a", "a", "c"], categories=["a", "c"]) |
44 |
| - tm.assert_index_equal(res, exp, exact=True) |
45 |
| - tm.assert_numpy_array_equal(indexer, np.array([0, 3, 2], dtype=np.intp)) |
| 36 | + with pytest.raises(ValueError, match=msg): |
| 37 | + ci.reindex(Categorical(["a", "c"])) |
46 | 38 |
|
47 | 39 | def test_reindex_duplicate_target(self):
|
48 | 40 | # See GH25459
|
|
0 commit comments