def test_remove_by_type(self):
RecordPrototype.create(type=relations.RECORD_TYPE.TEST_INT, date=self.date, value_int=41.7)
record = RecordPrototype.create(type=relations.RECORD_TYPE.TEST_FLOAT, date=self.date, value_float=66.6)
RecordPrototype.remove_by_type(relations.RECORD_TYPE.TEST_INT)
self.assertEqual(list(RecordPrototype._db_all().values_list('id', flat=True)), [record.id])
def test_select__int(self):
dates = [self.date + self.timedelta * i for i in range(4)]
RecordPrototype.create(type=relations.RECORD_TYPE.TEST_INT, date=dates[1], value_int=417)
RecordPrototype.create(type=relations.RECORD_TYPE.TEST_INT, date=dates[2], value_int=666)
self.assertEqual(
RecordPrototype.select(type=relations.RECORD_TYPE.TEST_INT, date_from=dates[0], date_to=dates[3]),
[(dates[1], 417), (dates[2], 666)],
)
self.assertEqual(
RecordPrototype.select(type=relations.RECORD_TYPE.TEST_INT, date_from=dates[1], date_to=dates[2]),
[(dates[1], 417), (dates[2], 666)],
)
self.assertEqual(
RecordPrototype.select(type=relations.RECORD_TYPE.TEST_INT, date_from=dates[1], date_to=dates[1]),
[(dates[1], 417)],
)
self.assertEqual(
RecordPrototype.select(type=relations.RECORD_TYPE.TEST_INT, date_from=dates[2], date_to=dates[2]),
[(dates[2], 666)],
)
self.assertEqual(
RecordPrototype.select(type=relations.RECORD_TYPE.TEST_INT, date_from=dates[3], date_to=dates[3]), []
)
def test_select__float(self):
date_1 = datetime.datetime.now()
date_2 = datetime.datetime.now()
date_3 = datetime.datetime.now()
date_4 = datetime.datetime.now()
RecordPrototype.create(type=relations.RECORD_TYPE.TEST_FLOAT, date=date_2, value_float=41.7)
RecordPrototype.create(type=relations.RECORD_TYPE.TEST_FLOAT, date=date_3, value_float=66.6)
self.assertEqual(RecordPrototype.select(type=relations.RECORD_TYPE.TEST_FLOAT, date_from=date_1, date_to=date_4), [(date_2, 41.7), (date_3, 66.6)])
self.assertEqual(RecordPrototype.select(type=relations.RECORD_TYPE.TEST_FLOAT, date_from=date_2, date_to=date_3), [(date_2, 41.7), (date_3, 66.6)])
self.assertEqual(RecordPrototype.select(type=relations.RECORD_TYPE.TEST_FLOAT, date_from=date_2, date_to=date_2), [(date_2, 41.7)])
self.assertEqual(RecordPrototype.select(type=relations.RECORD_TYPE.TEST_FLOAT, date_from=date_3, date_to=date_3), [(date_3, 66.6)])
self.assertEqual(RecordPrototype.select(type=relations.RECORD_TYPE.TEST_FLOAT, date_from=date_4, date_to=date_4), [])
def test_create(self):
with self.check_delta(RecordPrototype._db_count, 1):
record = RecordPrototype.create(type=relations.RECORD_TYPE.TEST_INT,
date=self.date,
value_int=666, value_float=666.6)
record.reload()
self.assertTrue(record.type.is_TEST_INT)
self.assertEqual(record.date, self.date)
self.assertEqual(record.value_int, 666)
self.assertEqual(record.value_float, 666.6)
def store_value(self, date, value):
return RecordPrototype.create(
type=self.TYPE,
date=date,
value_int=value if self.TYPE.value_type.is_INT else None,
value_float=value if self.TYPE.value_type.is_FLOAT else None)
def test_create__float_specified(self):
record = RecordPrototype.create(type=relations.RECORD_TYPE.TEST_INT, date=self.date, value_int=666)
record.reload()
self.assertEqual(record.value_float, 666.0)
def store_value(self, date, value):
return RecordPrototype.create(type=self.TYPE,
date=date,
value_int=value if self.TYPE.value_type.is_INT else None,
value_float=value if self.TYPE.value_type.is_FLOAT else None)
