Я написал некоторый код, который преобразует некоторые входные данные во фрейм данных Pandas, убедившись, что поля даты и времени преобразованы в соответствующий тип данных (Timestamp и Timedelta).
В моих тестах я пишу утверждения для dtypes столбцов, это выглядит так:
test_data_frame = input_data_to_df(test_data) # type: pd.DataFrame
self.assertEquals(
pandas.core.dtypes.dtype.DatetimeTZDtype.construct_from_string('datetime64[ns, UTC]')
test_data_frame['timestamp_column'].dtype)
Это хорошо работает для столбца типа datetime64[ns, UTC] или любого datetime64
Я пытаюсь сделать то же самое со столбцом типа timedelta64[ns], но не могу получить соответствующее представление dtype.
- Это не работает
DatetimeTZDtype.construct_from_string('timedelta64[ns]')
IntervalDtype.construct_from_string('timedelta64[ns]') возвращает interval[timedelta64[ns]], что не то, что я хочу