Skip to content

Commit 8a0266f

Browse files
committed
fixed test case issue
1 parent f0e3942 commit 8a0266f

File tree

1 file changed

+16
-8
lines changed

1 file changed

+16
-8
lines changed

pandas/tests/series/test_constructors.py

Lines changed: 16 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1883,23 +1883,31 @@ class A(OrderedDict):
18831883
(
18841884
{("a", "a"): 0.0, ("b", "a"): 1.0, ("b", "c"): 2.0, "z": 111.0},
18851885
None,
1886-
Index(["z", ("a", "a"), ("b", "a"), ("b", "c")], tupleize_cols=False),
1886+
Index(["z", ("a", "a"), ("b", "a"), ("b", "c")], dtype=object),
18871887
),
18881888
],
18891889
)
18901890
def test_constructor_dict_multiindex(
1891-
data, expected_index_multi, expected_index_single
1891+
self, data, expected_index_multi, expected_index_single
18921892
):
1893-
_d = sorted(data.items())
1893+
if all(isinstance(k, tuple) for k in data.keys()):
1894+
sorted_data = sorted(data.items())
1895+
else:
1896+
sorted_data = list(data.items())
18941897

18951898
result = Series(data)
1896-
if expected_index_multi:
1897-
expected = Series([x[1] for x in _d], index=expected_index_multi)
1899+
1900+
if expected_index_multi is not None:
1901+
expected = Series([x[1] for x in sorted_data], index=expected_index_multi)
18981902
tm.assert_series_equal(result, expected)
18991903

1900-
if expected_index_single:
1901-
result = result.reindex(index=expected_index_single)
1902-
expected = Series([x[1] for x in _d], index=expected_index_single)
1904+
if expected_index_single is not None:
1905+
result = result.reindex(index=expected_index_single, fill_value=np.nan)
1906+
expected_values = [
1907+
data[idx] if idx in data else np.nan for idx in expected_index_single
1908+
]
1909+
expected = Series(expected_values, index=expected_index_single)
1910+
19031911
tm.assert_series_equal(result, expected)
19041912

19051913
def test_constructor_dict_multiindex_reindex_flat(self):

0 commit comments

Comments
 (0)