28 lines
698 B
Python
28 lines
698 B
Python
|
import numpy as np
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from pandas import (
|
||
|
|
Index,
|
||
|
|
MultiIndex,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
# Note: identical the "multi" entry in the top-level "index" fixture
|
||
|
|
@pytest.fixture
|
||
|
|
def idx():
|
||
|
|
# a MultiIndex used to test the general functionality of the
|
||
|
|
# general functionality of this object
|
||
|
|
major_axis = Index(["foo", "bar", "baz", "qux"])
|
||
|
|
minor_axis = Index(["one", "two"])
|
||
|
|
|
||
|
|
major_codes = np.array([0, 0, 1, 2, 3, 3])
|
||
|
|
minor_codes = np.array([0, 1, 0, 1, 0, 1])
|
||
|
|
index_names = ["first", "second"]
|
||
|
|
mi = MultiIndex(
|
||
|
|
levels=[major_axis, minor_axis],
|
||
|
|
codes=[major_codes, minor_codes],
|
||
|
|
names=index_names,
|
||
|
|
verify_integrity=False,
|
||
|
|
)
|
||
|
|
return mi
|