def test_get_by_time(self):
upper = 10
lower = 0
check = 5
tmp_data_points = list()
for index in range(upper + 1):
tmp_data_point = BaseDataPoint({'glip': 'glop'})
tmp_data_points.append(tmp_data_point)
self.tmp_api.save_data_point(tmp_data_point)
# Upper bounds check
##################################################
start_timestamp = tmp_data_points[check].timestamp
end_timestamp = tmp_data_points[upper].timestamp
results = self.tmp_api.get_by_time(start_timestamp, end_timestamp)
self.assertIsInstance(results, list)
self.assertGreaterEqual(len(results), upper - check)
for result in results:
self.assertIsInstance(result, BaseDataPoint)
# Lower bounds check
##################################################
start_timestamp = tmp_data_points[lower].timestamp
end_timestamp = tmp_data_points[check].timestamp
results = self.tmp_api.get_by_time(start_timestamp, end_timestamp)
self.assertIsInstance(results, list)
self.assertGreaterEqual(len(results), check - lower)
for result in results:
self.assertIsInstance(result, BaseDataPoint)
def test_get_after(self):
check = 5
lower = 0
tmp_data_points = list()
for index in range(check + 1):
tmp_data_point = BaseDataPoint({'glip': 'glop'})
tmp_data_points.append(tmp_data_point)
self.tmp_api.save_data_point(tmp_data_point)
start_timestamp = tmp_data_points[lower].timestamp
results = self.tmp_api.get_after(start_timestamp)
self.assertIsInstance(results, list)
self.assertGreaterEqual(len(results), check)
for result in results:
self.assertIsInstance(result, BaseDataPoint)
def test_get_by_time_with_labels(self):
upper = 10
out = list()
labels = [self.tmp_label, self.tmp_api.all_label]
for _ in range(upper + 1):
tmp = BaseDataPoint({'glip': 'glop'})
out.append(tmp)
self.tmp_api.save_data_point(tmp, labels=labels)
results = self.tmp_api.get_by_time(out[0].timestamp,
out[-1].timestamp,
labels=labels)
self.assertIsInstance(results, list)
self.assertEqual(len(results), upper)
for result in results:
self.assertIsInstance(result, BaseDataPoint)
def test_get_before(self):
upper = 10
check = 5
tmp_data_points = list()
for index in range(upper + 1):
tmp_data_point = BaseDataPoint({'glip': 'glop'})
tmp_data_points.append(tmp_data_point)
self.tmp_api.save_data_point(tmp_data_point)
end_timestamp = tmp_data_points[check].timestamp
results = self.tmp_api.get_before(end_timestamp)
self.assertIsInstance(results, list)
# self.assertGreaterEqual(len(results), upper - check)
self.assertEqual(len(results), len(self.tmp_api.time_index) - check)
for result in results:
self.assertIsInstance(result, BaseDataPoint)
def setUp(self):
self.tmp_api = FileDataCatalog('./tmp')
self.tmp_label = Label({'name': 'thing'})
self.tmp_data_point = BaseDataPoint({'glip': 'glop'})