def test_plus_minus(self):
# Datetime + Datetime not supported
with self.assertRaises(TypeError) as cm:
self.dtvec1 + self.dtvec2
# Datetime slice -> Datetime
leading = self.dtvec1[1:]
trailing = self.dtvec1[:-1]
self.assertTrue(isinstance(leading, ak.Datetime) and isinstance(trailing, ak.Datetime))
# Datetime - Datetime -> Timedelta
diff = leading - trailing
self.assertTrue(isinstance(diff, ak.Timedelta))
self.assertTrue((diff == self.onesecond).all())
# Datetime - DatetimeScalar -> Timedelta
diff = self.dtvec1 - self.dtscalar
trange = ak.timedelta_range(start=0, periods=100, freq='s')
self.assertTrue(isinstance(diff, ak.Timedelta))
self.assertTrue((diff == trange).all())
# DatetimeScalar - Datetime -> Timedelta
diff = self.dtscalar - self.dtvec1
self.assertTrue(isinstance(diff, ak.Timedelta))
self.assertTrue((diff == (-trange)).all())
# Datetime + TimedeltaScalar -> Datetime
t = (trailing + self.onesecond)
self.assertTrue(isinstance(t, ak.Datetime))
self.assertTrue((t == leading).all())
# TimedeltaScalar + Datetime -> Datetime
t = (self.onesecond + trailing)
self.assertTrue(isinstance(t, ak.Datetime))
self.assertTrue((t == leading).all())
# Datetime - TimedeltaScalar -> Datetime
t = leading - self.onesecond
self.assertTrue(isinstance(t, ak.Datetime))
self.assertTrue((t == trailing).all())
# Datetime + Timedelta -> Datetime
t = (trailing + self.tdvec1[1:])
self.assertTrue(isinstance(t, ak.Datetime))
self.assertTrue((t == leading).all())
# Timedelta + Datetime -> Datetime
t = (self.tdvec1[1:] + trailing)
self.assertTrue(isinstance(t, ak.Datetime))
self.assertTrue((t == leading).all())
# Datetime - Timedelat -> Datetime
t = (leading - self.tdvec1[1:])
self.assertTrue(isinstance(t, ak.Datetime))
# Timedelta + Timedelta -> Timedelta
t = self.tdvec1 + self.tdvec1
self.assertTrue(isinstance(t, ak.Timedelta))
self.assertTrue((t == ak.Timedelta(2*ak.ones(100, dtype=ak.int64), unit='s')).all())
# Timedelta + TimedeltaScalar -> Timedelta
t = self.tdvec1 + self.onesecond
self.assertTrue(isinstance(t, ak.Timedelta))
self.assertTrue((t == ak.Timedelta(2*ak.ones(100, dtype=ak.int64), unit='s')).all())
# Timedelta - Timedelta -> Timedelta
t = self.tdvec1 - self.tdvec1
self.assertTrue(isinstance(t, ak.Timedelta))
self.assertTrue((t == ak.Timedelta(ak.zeros(100, dtype=ak.int64), unit='s')).all())
# Timedelta - TimedeltaScalar -> Timedelta
t = self.tdvec1 - self.onesecond
self.assertTrue(isinstance(t, ak.Timedelta))
self.assertTrue((t == ak.Timedelta(ak.zeros(100, dtype=ak.int64), unit='s')).all())
def test_timedel_std(self):
# pandas std uses unbiased estimator by default, in order to compare we set ddof=1 for arkouda std
ak_timedel_std = ak.Timedelta(ak.array([123, 456, 789]),
unit='s').std(ddof=1)
pd_timedel_std = pd.to_timedelta([123, 456, 789], unit='s').std()
self.assertEqual(ak_timedel_std, pd_timedel_std)
def setUp(self):
ArkoudaTest.setUp(self)
self.dtvec1 = ak.date_range(start='2021-01-01 12:00:00', periods=100, freq='s')
self.dtvec2 = ak.Datetime(pd.date_range('2021-01-01 12:00:00', periods=100, freq='s'))
self.dtscalar = pd.Timestamp('2021-01-01 12:00:00')
self.tdvec1 = ak.timedelta_range(start='1 second', end='1 second', periods=100)
self.tdvec2 = ak.Timedelta(ak.ones(100, dtype=ak.int64), unit='s')
self.onesecond = pd.Timedelta(1, unit='s')
def check_equal(pdunit, akunit):
pdval = pd.Timestamp(1, unit=pdunit)
akval = ak.Datetime(ak.ones(10, dtype=ak.int64), unit=akunit)[0]
try:
self.assertEqual(pdval, akval)
except AssertionError:
logging.getLogger().error("pandas {} ({}) != arkouda {} ({})".format(pdunit, pdval, akunit, akval))
pdval = pd.Timedelta(1, unit=pdunit)
akval = ak.Timedelta(ak.ones(10, dtype=ak.int64), unit=akunit)[0]
try:
self.assertEqual(pdval, akval)
except AssertionError:
logging.getLogger().error("pandas {} ({}) != arkouda {} ({})".format(pdunit, pdval, akunit, akval))
def test_noop_creation(self):
self.assertTrue((ak.Datetime(self.dtvec1) == self.dtvec1).all())
self.assertTrue((ak.Timedelta(self.tdvec1) == self.tdvec1).all())
