Skip to content

Commit 5ed1a0b

Browse files
authored
CLN: organize MultiIndex indexing tests (#31863)
* implement test_take * move test_take, organize tests * cleanup * remove unused imports * isort fixup * typo fixup
1 parent 9aa9e4f commit 5ed1a0b

File tree

2 files changed

+350
-362
lines changed

2 files changed

+350
-362
lines changed

pandas/conftest.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -956,6 +956,25 @@ def __len__(self):
956956
return TestNonDictMapping
957957

958958

959+
def _gen_mi():
960+
# a MultiIndex used to test the general functionality of this object
961+
962+
# See Also: tests.multi.conftest.idx
963+
major_axis = Index(["foo", "bar", "baz", "qux"])
964+
minor_axis = Index(["one", "two"])
965+
966+
major_codes = np.array([0, 0, 1, 2, 3, 3])
967+
minor_codes = np.array([0, 1, 0, 1, 0, 1])
968+
index_names = ["first", "second"]
969+
mi = MultiIndex(
970+
levels=[major_axis, minor_axis],
971+
codes=[major_codes, minor_codes],
972+
names=index_names,
973+
verify_integrity=False,
974+
)
975+
return mi
976+
977+
959978
indices_dict = {
960979
"unicode": tm.makeUnicodeIndex(100),
961980
"string": tm.makeStringIndex(100),
@@ -972,6 +991,7 @@ def __len__(self):
972991
"interval": tm.makeIntervalIndex(100),
973992
"empty": Index([]),
974993
"tuples": MultiIndex.from_tuples(zip(["foo", "bar", "baz"], [1, 2, 3])),
994+
"multi": _gen_mi(),
975995
"repeats": Index([0, 0, 1, 1, 2, 2]),
976996
}
977997

0 commit comments

Comments
 (0)