Skip to content

Commit 1572b3f

Browse files
authored
TST: Add test_datetime_object_multiindex test (#52913)
add test_datetime_object_multiindex test
1 parent ad23226 commit 1572b3f

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed

pandas/tests/test_multilevel.py

+24
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
import datetime
2+
13
import numpy as np
24
import pytest
35

@@ -266,6 +268,28 @@ def test_subsets_multiindex_dtype(self):
266268
result = df.a.b.dtypes
267269
tm.assert_series_equal(result, expected)
268270

271+
def test_datetime_object_multiindex(self):
272+
data_dic = {
273+
(0, datetime.date(2018, 3, 3)): {"A": 1, "B": 10},
274+
(0, datetime.date(2018, 3, 4)): {"A": 2, "B": 11},
275+
(1, datetime.date(2018, 3, 3)): {"A": 3, "B": 12},
276+
(1, datetime.date(2018, 3, 4)): {"A": 4, "B": 13},
277+
}
278+
result = DataFrame.from_dict(data_dic, orient="index")
279+
data = {"A": [1, 2, 3, 4], "B": [10, 11, 12, 13]}
280+
index = [
281+
[0, 0, 1, 1],
282+
[
283+
datetime.date(2018, 3, 3),
284+
datetime.date(2018, 3, 4),
285+
datetime.date(2018, 3, 3),
286+
datetime.date(2018, 3, 4),
287+
],
288+
]
289+
expected = DataFrame(data=data, index=index)
290+
291+
tm.assert_frame_equal(result, expected)
292+
269293

270294
class TestSorted:
271295
"""everything you wanted to test about sorting"""

0 commit comments

Comments
 (0)