Python is_datetime64_arrayの例

コード例 #1

    def test_is_datetimelike_array_all_nan_nat_like(self):
        arr = np.array([np.nan, pd.NaT, np.datetime64('nat')])
        self.assertTrue(lib.is_datetime_array(arr))
        self.assertTrue(lib.is_datetime64_array(arr))
        self.assertFalse(lib.is_timedelta_array(arr))
        self.assertFalse(lib.is_timedelta64_array(arr))
        self.assertFalse(lib.is_timedelta_or_timedelta64_array(arr))

        arr = np.array([np.nan, pd.NaT, np.timedelta64('nat')])
        self.assertFalse(lib.is_datetime_array(arr))
        self.assertFalse(lib.is_datetime64_array(arr))
        self.assertTrue(lib.is_timedelta_array(arr))
        self.assertTrue(lib.is_timedelta64_array(arr))
        self.assertTrue(lib.is_timedelta_or_timedelta64_array(arr))

        arr = np.array(
            [np.nan, pd.NaT,
             np.datetime64('nat'),
             np.timedelta64('nat')])
        self.assertFalse(lib.is_datetime_array(arr))
        self.assertFalse(lib.is_datetime64_array(arr))
        self.assertFalse(lib.is_timedelta_array(arr))
        self.assertFalse(lib.is_timedelta64_array(arr))
        self.assertFalse(lib.is_timedelta_or_timedelta64_array(arr))

        arr = np.array([np.nan, pd.NaT])
        self.assertTrue(lib.is_datetime_array(arr))
        self.assertTrue(lib.is_datetime64_array(arr))
        self.assertTrue(lib.is_timedelta_array(arr))
        self.assertTrue(lib.is_timedelta64_array(arr))
        self.assertTrue(lib.is_timedelta_or_timedelta64_array(arr))

        arr = np.array([np.nan, np.nan], dtype=object)
        self.assertFalse(lib.is_datetime_array(arr))
        self.assertFalse(lib.is_datetime64_array(arr))
        self.assertFalse(lib.is_timedelta_array(arr))
        self.assertFalse(lib.is_timedelta64_array(arr))
        self.assertFalse(lib.is_timedelta_or_timedelta64_array(arr))

コード例 #2

ファイル: test_inference.py プロジェクト: cgrin/pandas

    def test_is_datetimelike_array_all_nan_nat_like(self):
        arr = np.array([np.nan, pd.NaT, np.datetime64('nat')])
        self.assertTrue(lib.is_datetime_array(arr))
        self.assertTrue(lib.is_datetime64_array(arr))
        self.assertFalse(lib.is_timedelta_array(arr))
        self.assertFalse(lib.is_timedelta64_array(arr))
        self.assertFalse(lib.is_timedelta_or_timedelta64_array(arr))

        arr = np.array([np.nan, pd.NaT, np.timedelta64('nat')])
        self.assertFalse(lib.is_datetime_array(arr))
        self.assertFalse(lib.is_datetime64_array(arr))
        self.assertTrue(lib.is_timedelta_array(arr))
        self.assertTrue(lib.is_timedelta64_array(arr))
        self.assertTrue(lib.is_timedelta_or_timedelta64_array(arr))

        arr = np.array([np.nan, pd.NaT, np.datetime64('nat'),
                        np.timedelta64('nat')])
        self.assertFalse(lib.is_datetime_array(arr))
        self.assertFalse(lib.is_datetime64_array(arr))
        self.assertFalse(lib.is_timedelta_array(arr))
        self.assertFalse(lib.is_timedelta64_array(arr))
        self.assertFalse(lib.is_timedelta_or_timedelta64_array(arr))

        arr = np.array([np.nan, pd.NaT])
        self.assertTrue(lib.is_datetime_array(arr))
        self.assertTrue(lib.is_datetime64_array(arr))
        self.assertTrue(lib.is_timedelta_array(arr))
        self.assertTrue(lib.is_timedelta64_array(arr))
        self.assertTrue(lib.is_timedelta_or_timedelta64_array(arr))

        arr = np.array([np.nan, np.nan], dtype=object)
        self.assertFalse(lib.is_datetime_array(arr))
        self.assertFalse(lib.is_datetime64_array(arr))
        self.assertFalse(lib.is_timedelta_array(arr))
        self.assertFalse(lib.is_timedelta64_array(arr))
        self.assertFalse(lib.is_timedelta_or_timedelta64_array(arr))