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'})