def setUp(self):
super().setUp()
self.fixture = DataPoint()
self.fixture.set_name("datapoint")
self.fixture.set_value(0.0)
self.fixture.set_timestamp(int(datetime.timestamp(datetime.now())))
self.db.add(self.fixture)
self.db.commit()
def test_data(self):
dp = DataPoint(name="datapoint",
value=1.0,
timestamp=int(datetime.timestamp(datetime.now())))
self.db.add(dp)
self.fixture.add_data_point(dp)
self.db.commit()
self.assertIn(dp, self.fixture.data, "point not contained")
def test_know_data_point(self):
dp = DataPoint(name="datapoint",
value=1.0,
timestamp=int(datetime.timestamp(datetime.now())))
self.db.add(dp)
self.fixture.add_data_point(dp)
count = self.fixture.data.count()
self.fixture.add_data_point(dp)
self.assertEqual(count, self.fixture.data.count(),
"counts are not the same")
def _create_data_point(self, name, value, ts):
"""Constructs a data point with the given value and timestamp and commit the changes to the database
Args:
**name (str)**: the name of the data point
**value (float)**: the measured value
**ts (int)**: the timestamp
Returns:
the newly created DataPoint
"""
dp = DataPoint(name=name, value=value, timestamp=ts)
self.db.add(dp)
sensor = self.db.query(Sensor).filter_by(name=self._sensor_name).first()
sensor.add_data_point(dp)
self.db.commit()
return dp
def create_point(json, sensor):
point = DataPoint(json)
current_app.session.add(point)
current_app.session.commit()
sensor.add_data_point(point)
current_app.session.commit()
def test_create_from_json(self):
exp = {"id": 2, "name": "datapoint", "value": 2.0, "timestamp": 10}
self.fixture = DataPoint(exp)
self.assertEqual(exp['name'], self.fixture.get_name())
self.assertAlmostEqual(exp['value'], self.fixture.get_value(), 3)
self.assertEqual(exp['timestamp'], self.fixture.get_timestamp())
class DataPointTest(BaseTest):
def setUp(self):
super().setUp()
self.fixture = DataPoint()
self.fixture.set_name("datapoint")
self.fixture.set_value(0.0)
self.fixture.set_timestamp(int(datetime.timestamp(datetime.now())))
self.db.add(self.fixture)
self.db.commit()
def test_id(self):
self.assertEqual(self.fixture.get_id(), 1, "id's do not match")
def test_name(self):
self.assertEqual(self.fixture.get_name(), "datapoint")
def test_null_name(self):
with self.assertRaises(Exception):
self.fixture.set_name(None)
def test_nonstring_name(self):
with self.assertRaises(Exception):
self.fixture.set_name(3)
def test_value(self):
self.assertAlmostEqual(self.fixture.get_value(), 0, 3,
"values do not match")
def test_null_value(self):
with self.assertRaises(Exception):
self.fixture.set_value(None)
def test_nonfloat_value(self):
with self.assertRaises(Exception):
self.fixture.set_value("Test")
def test_parent_sensor(self):
sens = Sensor(name="TestSensor", quantity="Power", unit=Unit.kWh)
self.db.add(sens)
sens.add_data_point(self.fixture)
self.db.commit()
self.assertEqual(self.fixture.sensor_id, sens.get_id(), "id mismatch")
def test_timestamp(self):
self.assertEqual(int(datetime.timestamp(datetime.now())),
self.fixture.get_timestamp(),
"timestamps do not match")
def test_non_int_timestamp(self):
with self.assertRaises(Exception):
self.fixture.set_timestamp("001")
def test_null_timestamp(self):
with self.assertRaises(Exception):
self.fixture.set_timestamp(None)
def test_repr(self):
self.assertEqual("Data Point: 0.0", self.fixture.__repr__(),
"repr not same")
def test_str(self):
self.assertEqual("Data Point: 0.0", str(self.fixture),
"string representation not same")
def test_to_dict(self):
exp = {
"id": 1,
"name": "datapoint",
"value": 0.0,
"timestamp": int(datetime.timestamp(datetime.now()))
}
self.assertDictEqual(exp, self.fixture.to_dict(), "dicts not the same")
def test_update(self):
exp = {
"id": 1,
"name": "updated_datapoint",
"value": 2.0,
"timestamp": 10
}
self.fixture.update(exp)
self.assertEqual(exp['id'], self.fixture.get_id())
self.assertEqual(exp['name'], self.fixture.get_name())
self.assertAlmostEqual(exp['value'], self.fixture.get_value(), 3)
self.assertEqual(exp['timestamp'], self.fixture.get_timestamp())
def test_create_from_json(self):
exp = {"id": 2, "name": "datapoint", "value": 2.0, "timestamp": 10}
self.fixture = DataPoint(exp)
self.assertEqual(exp['name'], self.fixture.get_name())
self.assertAlmostEqual(exp['value'], self.fixture.get_value(), 3)
self.assertEqual(exp['timestamp'], self.fixture.get_timestamp())