def test_init_dataframe(self):
edata = self.data
inst = Aouda(dataframe=self.data)
self.assertTrue(
inst.data.eq(edata).all().all(),
'exp: %s; act: %s' % (edata, inst.data))
def test_get_data_shift_data(self):
edata = self.data.shift(-10, 'T')
inst = Aouda(dataframe=edata, shift_data=True)
period = 2
data, until = inst.get_data(period)
delta = (until - edata.index[0]).total_seconds()
edata = edata.shift(periods=int(delta), freq='S')
since = until - timedelta(seconds=period)
edata = edata[edata.index >= since][edata.index <= until]
self.assertEqual(len(edata), len(data),
'exp: %s; act: %s' % (len(edata), len(data)))
for i in range(len(data)):
self.assertEqual(
data[i].timestamp, edata.index[i],
'exp: %s;\nact: %s' % (data[i].timestamp, edata.index[i]))
self.assertEqual(
data[i].hr, edata.ix[i]['hr'],
'exp: %s; act: %s' % (data[i].hr, edata.ix[i]['hr']))
self.assertEqual(
data[i].acc_x, edata.ix[i]['acc_x'],
'exp: %s; act: %s' % (data[i].acc_x, edata.ix[i]['acc_x']))
self.assertEqual(
data[i].acc_y, edata.ix[i]['acc_y'],
'exp: %s; act: %s' % (data[i].acc_y, edata.ix[i]['acc_y']))
self.assertEqual(
data[i].acc_z, edata.ix[i]['acc_z'],
'exp: %s; act: %s' % (data[i].acc_z, edata.ix[i]['acc_z']))
def test_get_data(self):
inst = Aouda(dataframe=self.data)
period = 2
data, until = inst.get_data(period)
since = until - timedelta(seconds=period)
edata = self.data[self.data.index >= since][self.data.index <= until]
self.assertEqual(len(edata), len(data),
'exp: %s; act: %s' % (len(edata), len(data)))
for i in range(len(data)):
self.assertEqual(
data[i].timestamp, edata.index[i],
'exp: %s; act: %s' % (data[i].timestamp, edata.index[i]))
self.assertEqual(
data[i].hr, edata.ix[i]['hr'],
'exp: %s; act: %s' % (data[i].hr, edata.ix[i]['hr']))
self.assertEqual(
data[i].acc_x, edata.ix[i]['acc_x'],
'exp: %s; act: %s' % (data[i].acc_x, edata.ix[i]['acc_x']))
self.assertEqual(
data[i].acc_y, edata.ix[i]['acc_y'],
'exp: %s; act: %s' % (data[i].acc_y, edata.ix[i]['acc_y']))
self.assertEqual(
data[i].acc_z, edata.ix[i]['acc_z'],
'exp: %s; act: %s' % (data[i].acc_z, edata.ix[i]['acc_z']))
def test_read_acc_magn(self):
inst = Aouda(dataframe=self.data)
until = datetime.now()
acc_magn = inst.read_acc_magn()
eacc_magn = self.data[self.data.index <= until]['acc'][-1]
self.assertEqual(acc_magn, eacc_magn,
'exp: %s; act: %s' % (eacc_magn, acc_magn))
def test_read_heart_rate(self):
inst = Aouda(dataframe=self.data)
until = datetime.now()
heart_rate = inst.read_heart_rate()
eheart_rate = self.data[self.data.index <= until]['hr'][-1]
self.assertEqual(heart_rate, eheart_rate,
'exp: %s; act: %s' % (eheart_rate, heart_rate))
def test_init_filename(self):
edata = self.data
inst = Aouda(filename=AoudaTests.filename,
base_datetime=AoudaTests.base_datetime)
self.assertEqual(len(inst.data), len(edata),
'exp: %s; act: %s' % (len(inst.data), len(edata)))
for i in range(len(inst.data)):
self.assertItemsEqual(
inst.data.ix[i].tolist(), edata.ix[i].tolist(),
'exp: %s; act: %s' %
(edata.ix[i].tolist(), inst.data.ix[i].tolist()))
def test_get_data_no_more_data(self):
inst = Aouda(dataframe=self.data.shift(-10, 'T'))
self.assertRaises(NoDataAvailableError, inst.get_data, 2)
def test_read_acc_magn_no_enough_data(self):
inst = Aouda(dataframe=self.data.shift(10, 'T'))
acc_magn = inst.read_acc_magn()
self.assertTrue(isnan(acc_magn))
def test_read_heart_rate_no_enough_data(self):
inst = Aouda(dataframe=self.data.shift(10, 'T'))
heart_rate = inst.read_heart_rate()
self.assertTrue(isnan(heart_rate))