@@ -956,6 +956,25 @@ def __len__(self):
956
956
return TestNonDictMapping
957
957
958
958
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
+
959
978
indices_dict = {
960
979
"unicode" : tm .makeUnicodeIndex (100 ),
961
980
"string" : tm .makeStringIndex (100 ),
@@ -972,6 +991,7 @@ def __len__(self):
972
991
"interval" : tm .makeIntervalIndex (100 ),
973
992
"empty" : Index ([]),
974
993
"tuples" : MultiIndex .from_tuples (zip (["foo" , "bar" , "baz" ], [1 , 2 , 3 ])),
994
+ "multi" : _gen_mi (),
975
995
"repeats" : Index ([0 , 0 , 1 , 1 , 2 , 2 ]),
976
996
}
977
997
0 commit comments