Skip to content

Commit 906b51a

Browse files
mroeschkejorisvandenbossche
authored andcommitted
TST: Test datetime array assignment with different units (pandas-dev#7492) (pandas-dev#14884)
1 parent e80a2b9 commit 906b51a

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

pandas/tests/frame/test_timeseries.py

+15
Original file line numberDiff line numberDiff line change
@@ -366,6 +366,21 @@ def test_operation_on_NaT(self):
366366
exp = pd.Series([pd.NaT], index=["foo"])
367367
tm.assert_series_equal(res, exp)
368368

369+
def test_datetime_assignment_with_NaT_and_diff_time_units(self):
370+
# GH 7492
371+
data_ns = np.array([1, 'nat'], dtype='datetime64[ns]')
372+
result = pd.Series(data_ns).to_frame()
373+
result['new'] = data_ns
374+
expected = pd.DataFrame({0: [1, None],
375+
'new': [1, None]}, dtype='datetime64[ns]')
376+
tm.assert_frame_equal(result, expected)
377+
# OutOfBoundsDatetime error shouldn't occur
378+
data_s = np.array([1, 'nat'], dtype='datetime64[s]')
379+
result['new'] = data_s
380+
expected = pd.DataFrame({0: [1, None],
381+
'new': [1e9, None]}, dtype='datetime64[ns]')
382+
tm.assert_frame_equal(result, expected)
383+
369384

370385
if __name__ == '__main__':
371386
import nose

0 commit comments

Comments
 (0)