|
| 1 | +import datetime |
| 2 | + |
1 | 3 | import numpy as np
|
2 | 4 | import pytest
|
3 | 5 |
|
@@ -266,6 +268,28 @@ def test_subsets_multiindex_dtype(self):
|
266 | 268 | result = df.a.b.dtypes
|
267 | 269 | tm.assert_series_equal(result, expected)
|
268 | 270 |
|
| 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 | + |
269 | 293 |
|
270 | 294 | class TestSorted:
|
271 | 295 | """everything you wanted to test about sorting"""
|
|
0 commit comments