File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -371,12 +371,14 @@ def test_datetime_assignment_with_NaT_and_diff_time_units(self):
371
371
data_ns = np .array ([1 , 'nat' ], dtype = 'datetime64[ns]' )
372
372
result = pd .Series (data_ns ).to_frame ()
373
373
result ['new' ] = data_ns
374
- expected = pd .DataFrame ({0 : data_ns , 'new' : data_ns })
374
+ expected = pd .DataFrame ({0 : [1 , None ],
375
+ 'new' : [1 , None ]}, dtype = 'datetime64[ns]' )
375
376
tm .assert_frame_equal (result , expected )
376
377
# OutOfBoundsDatetime error shouldn't occur
377
378
data_s = np .array ([1 , 'nat' ], dtype = 'datetime64[s]' )
378
379
result ['new' ] = data_s
379
- expected = pd .DataFrame ({0 : data_ns , 'new' : data_s })
380
+ expected = pd .DataFrame ({0 : [1 , None ],
381
+ 'new' : [1e9 , None ]}, dtype = 'datetime64[ns]' )
380
382
tm .assert_frame_equal (result , expected )
381
383
382
384
You can’t perform that action at this time.
0 commit comments