Handle datetime64 and timedelta64

Author: TomAugspurger

pandas/core/arrays/array_.py

@@ -225,14 +225,16 @@ def array(data,         # type: Sequence[object]
                 # We choose to return an ndarray, rather than raising.
                 pass
 
-        elif inferred_dtype == 'datetime':
+        elif inferred_dtype.startswith('datetime'):
+            # datetime, datetime64
             try:
                 return DatetimeArrayMixin._from_sequence(data, copy=copy)
             except ValueError:
                 # Mixture of timezones, fall back to PandasArray
                 pass
 
-        elif inferred_dtype == 'timedelta':
+        elif inferred_dtype.startswith('timedelta'):
+            # timedelta, timedelta64
             return TimedeltaArrayMixin._from_sequence(data, copy=copy)
 
         # TODO(BooleanArray): handle this type

pandas/tests/arrays/test_array.py

@@ -110,6 +110,12 @@ def test_array_copy():
     ([datetime.datetime(2000, 1, 1), datetime.datetime(2001, 1, 1)],
      pd.arrays.DatetimeArray._from_sequence(['2000', '2001'])),
 
+    (np.array([1, 2], dtype='M8[ns]'),
+     pd.arrays.DatetimeArray(np.array([1, 2], dtype='M8[ns]'))),
+
+    (np.array([1, 2], dtype='M8[us]'),
+     pd.arrays.DatetimeArray(np.array([1000, 2000], dtype='M8[ns]'))),
+
     # datetimetz
     ([pd.Timestamp('2000', tz='CET'), pd.Timestamp('2001', tz='CET')],
      pd.arrays.DatetimeArray._from_sequence(
@@ -123,6 +129,13 @@ def test_array_copy():
     # timedelta
     ([pd.Timedelta('1H'), pd.Timedelta('2H')],
      pd.arrays.TimedeltaArray._from_sequence(['1H', '2H'])),
+
+    (np.array([1, 2], dtype='m8[ns]'),
+     pd.arrays.TimedeltaArray(np.array([1, 2], dtype='m8[ns]'))),
+
+    (np.array([1, 2], dtype='m8[us]'),
+     pd.arrays.TimedeltaArray(np.array([1000, 2000], dtype='m8[ns]'))),
+
 ])
 def test_array_inference(data, expected):
     result = pd.array(data)