Skip to content

Commit d71f8ac

Browse files
committed
simplify test
1 parent cadbdc3 commit d71f8ac

File tree

1 file changed

+6
-60
lines changed

1 file changed

+6
-60
lines changed

pandas/tests/test_multilevel.py

+6-60
Original file line numberDiff line numberDiff line change
@@ -1934,66 +1934,12 @@ def test_repeat(self):
19341934

19351935
def test_subsets_multiindex_dtype(self):
19361936
# GH 20757
1937-
data = [
1938-
[
1939-
"n",
1940-
1,
1941-
0,
1942-
False,
1943-
2,
1944-
1,
1945-
False,
1946-
0,
1947-
0,
1948-
False,
1949-
2,
1950-
0,
1951-
False,
1952-
0,
1953-
1,
1954-
False,
1955-
1,
1956-
1,
1957-
False,
1958-
"o",
1959-
1521734085.289453,
1960-
"p",
1961-
3233,
1962-
1521734085.289494,
1963-
]
1964-
]
1965-
1966-
columns = [
1967-
("a", "d", "i", np.nan, np.nan),
1968-
("a", "d", "j", 0.0, "k"),
1969-
("a", "d", "j", 0.0, "l"),
1970-
("a", "d", "j", 0.0, "m"),
1971-
("a", "d", "j", 1.0, "k"),
1972-
("a", "d", "j", 1.0, "l"),
1973-
("a", "d", "j", 1.0, "m"),
1974-
("a", "d", "j", 2.0, "k"),
1975-
("a", "d", "j", 2.0, "l"),
1976-
("a", "d", "j", 2.0, "m"),
1977-
("a", "d", "j", 3.0, "k"),
1978-
("a", "d", "j", 3.0, "l"),
1979-
("a", "d", "j", 3.0, "m"),
1980-
("a", "d", "j", 4.0, "k"),
1981-
("a", "d", "j", 4.0, "l"),
1982-
("a", "d", "j", 4.0, "m"),
1983-
("a", "d", "j", 5.0, "k"),
1984-
("a", "d", "j", 5.0, "l"),
1985-
("a", "d", "j", 5.0, "m"),
1986-
("b", "f", np.nan, np.nan, np.nan),
1987-
("b", "h", np.nan, np.nan, np.nan),
1988-
("c", "e", np.nan, np.nan, np.nan),
1989-
("c", "g", np.nan, np.nan, np.nan),
1990-
("c", "h", np.nan, np.nan, np.nan),
1991-
]
1992-
1993-
columns = pd.MultiIndex.from_tuples(columns)
1994-
df = pd.DataFrame(data, columns=columns)
1995-
1996-
tm.assert_equal(df.dtypes.a.d.i, df.a.d.i.dtypes)
1937+
data = [["x", 1]]
1938+
columns = [("a", "b", np.nan), ("a", "c", 0.0)]
1939+
df = DataFrame(data, columns=pd.MultiIndex.from_tuples(columns))
1940+
expected = df.dtypes.a.b
1941+
result = df.a.b.dtypes
1942+
tm.assert_series_equal(expected, result)
19971943

19981944

19991945
class TestSorted(Base):

0 commit comments

Comments
 (0)