def astype(self, dtype, copy=True):
dtype = np.dtype(dtype)
if is_object_dtype(dtype):
return self.asobject
elif is_timedelta64_ns_dtype(dtype):
if copy is True:
return self.copy()
return self
elif is_timedelta64_dtype(dtype):
# return an index (essentially this is division)
result = self.values.astype(dtype, copy=copy)
if self.hasnans:
return Index(self._maybe_mask_results(result, convert="float64"), name=self.name)
return Index(result.astype("i8"), name=self.name)
elif is_integer_dtype(dtype):
return Index(self.values.astype("i8", copy=copy), dtype="i8", name=self.name)
raise ValueError("Cannot cast TimedeltaIndex to dtype %s" % dtype)
def test_is_timedelta(self):
self.assertTrue(is_timedelta64_dtype('timedelta64'))
self.assertTrue(is_timedelta64_dtype('timedelta64[ns]'))
self.assertFalse(is_timedelta64_ns_dtype('timedelta64'))
self.assertTrue(is_timedelta64_ns_dtype('timedelta64[ns]'))
tdi = TimedeltaIndex([1e14, 2e14], dtype='timedelta64')
self.assertTrue(is_timedelta64_dtype(tdi))
self.assertTrue(is_timedelta64_ns_dtype(tdi))
self.assertTrue(is_timedelta64_ns_dtype(tdi.astype('timedelta64[ns]')))
# Conversion to Int64Index:
self.assertFalse(is_timedelta64_ns_dtype(tdi.astype('timedelta64')))
self.assertFalse(is_timedelta64_ns_dtype(tdi.astype('timedelta64[h]')))
def test_is_timedelta(self):
self.assertTrue(is_timedelta64_dtype('timedelta64'))
self.assertTrue(is_timedelta64_dtype('timedelta64[ns]'))
self.assertFalse(is_timedelta64_ns_dtype('timedelta64'))
self.assertTrue(is_timedelta64_ns_dtype('timedelta64[ns]'))
tdi = TimedeltaIndex([1e14, 2e14], dtype='timedelta64')
self.assertTrue(is_timedelta64_dtype(tdi))
self.assertTrue(is_timedelta64_ns_dtype(tdi))
self.assertTrue(is_timedelta64_ns_dtype(tdi.astype('timedelta64[ns]')))
# Conversion to Int64Index:
self.assertFalse(is_timedelta64_ns_dtype(tdi.astype('timedelta64')))
self.assertFalse(is_timedelta64_ns_dtype(tdi.astype('timedelta64[h]')))
def astype(self, dtype, copy=True):
dtype = np.dtype(dtype)
if is_object_dtype(dtype):
return self.asobject
elif is_timedelta64_ns_dtype(dtype):
if copy is True:
return self.copy()
return self
elif is_timedelta64_dtype(dtype):
# return an index (essentially this is division)
result = self.values.astype(dtype, copy=copy)
if self.hasnans:
return Index(self._maybe_mask_results(result,
convert='float64'),
name=self.name)
return Index(result.astype('i8'), name=self.name)
elif is_integer_dtype(dtype):
return Index(self.values.astype('i8', copy=copy), dtype='i8',
name=self.name)
raise ValueError('Cannot cast TimedeltaIndex to dtype %s' % dtype)
