def test_should_receive_data_from_attached_sensor():
snapper = Snapper()
sensor = Sensor()
snapper.add_sensor(sensor)
sensor.publish(2)
assert snapper.dataBuffer[sensor.uuid] == [2]
def __init__(self):
self.sensors = []
self.route_map = {}
self.reverse_route_map = {}
self.variables = []
self.snapper = Snapper(self)
self.matrix = AssociationMatrix()
def test_should_generate_and_return_multiple_snapshots_correctly():
snapper = Snapper()
sensor_a = Sensor()
snapper.add_sensor(sensor_a)
sensor_a.publish(1, 1)
sensor_a.publish(5, 6)
sensor_b = Sensor()
snapper.add_sensor(sensor_b)
sensor_b.publish(3, 7)
sensor_b.publish(9, 8)
snapshot1 = snapper.get_snapshot()
assert round(snapshot1[sensor_a.uuid] - 3, 5) == 0
assert round(snapshot1[sensor_b.uuid] - 6, 5) == 0
sensor_a.publish(5, 11)
sensor_a.publish(10, 12)
sensor_a.publish(15, 13)
sensor_b.publish(6, 13)
sensor_b.publish(12, 14)
sensor_b.publish(2, 15)
sensor_b.publish(30, 16)
snapshot2 = snapper.get_snapshot()
assert round(snapshot2[sensor_a.uuid] - 10, 5) == 0
assert round(snapshot2[sensor_b.uuid] - 12.5, 5) == 0
def test_aggregation_averaging():
snapper = Snapper()
sensor_a = Sensor()
snapper.add_sensor(sensor_a)
sensor_b = Sensor()
snapper.add_sensor(sensor_b)
sensor_a.publish(1, 1)
sensor_a.publish(5, 2)
sensor_a.publish(6, 2)
sensor_a.publish(4, 2)
sensor_b.publish(3, 2)
sensor_b.publish(7, 2)
sensor_b.publish(2, 2)
sensor_b.publish(4, 2)
sensor_b.publish(5, 2)
sensor_b.publish(2, 2)
sensor_b.publish(6, 2)
sensor_b.publish(9, 2)
snapshot = snapper.get_snapshot()
assert round(snapshot[sensor_a.uuid] - 4, 5) == 0
assert round(snapshot[sensor_b.uuid] - 4.75, 5) == 0
def test_should_remove_sensor():
snapper = Snapper()
sensor = Sensor()
snapper.add_sensor(sensor)
snapper.remove_sensor(sensor)
assert snapper.sensors == []
def test_should_generate_and_return_snapshot():
snapper = Snapper()
sensor_a = Sensor()
snapper.add_sensor(sensor_a)
sensor_a.publish(1)
sensor_b = Sensor()
snapper.add_sensor(sensor_b)
sensor_b.publish(3)
snapshot = snapper.get_snapshot()
assert round(snapshot[sensor_a.uuid] - 1, 5) == 0
assert round(snapshot[sensor_b.uuid] - 3, 5) == 0
def test_should_change_window_size():
snapper = Snapper()
snapper.set_window_size(20)
assert snapper.timeWindow == 20
snapper.set_window_size(5.00)
assert round(snapper.timeWindow - 5.00, 5) == 0.0
def test_should_use_none_for_sensors_not_pushing():
snapper = Snapper()
sensor_a = Sensor()
snapper.add_sensor(sensor_a)
sensor_b = Sensor()
snapper.add_sensor(sensor_b)
sensor_b.publish(3)
snapshot = snapper.get_snapshot()
assert sensor_a.uuid in snapshot
assert snapshot[sensor_a.uuid] is None
assert round(snapshot[sensor_b.uuid] - 3, 5) == 0
def test_should_use_changed_time_window_for_snapshot_after_change():
manager = Manager()
manager.on_data = MagicMock()
snapper = Snapper(manager)
sensor1 = Sensor()
snapper.add_sensor(sensor1)
sensor1.publish(2, 1)
# Change window size in middle of data transmission
snapper.set_window_size(15.0)
sensor2 = Sensor()
snapper.add_sensor(sensor2)
sensor2.publish(3, 1)
# Force a snapshot publish by pushing value with time past original window
sensor2.publish(3, 11)
# Should finish original window and push the resulting snapshot before
# transitioning
manager.on_data.assert_called_with({
'start': 1,
'end': 10,
sensor1.uuid: 2,
sensor2.uuid: 3
})
# Push values within the new window
sensor1.publish(3, 15)
sensor2.publish(5, 16)
# Force a snapshot push by publishing out of bounds data
sensor1.publish(3, 26)
manager.on_data.assert_called_with({
'start': 10,
'end': 24,
sensor1.uuid: 3,
sensor2.uuid: 4
})
def test_should_receive_data_from_multiple_sensors():
snapper = Snapper()
sensor1 = Sensor()
snapper.add_sensor(sensor1)
sensor1.publish(2)
sensor2 = Sensor()
snapper.add_sensor(sensor2)
startTime = snapper.windowStart
endTime = snapper.windowEnd
sensor2.publish(3)
assert snapper.dataBuffer == {sensor1.uuid: [2], sensor2.uuid: [3]}
assert snapper.get_snapshot() == {
"start": startTime,
"end": endTime,
sensor1.uuid: 2,
sensor2.uuid: 3
}
def test_should_properly_generate_snapshot_from_complete_data():
snapper = Snapper()
sensor1 = Sensor()
snapper.add_sensor(sensor1)
sensor2 = Sensor()
snapper.add_sensor(sensor2)
sensor1.publish(2)
assert snapper.snapshot == {}
startTime = snapper.windowStart
endTime = snapper.windowEnd
sensor2.publish(3)
assert snapper.get_snapshot() == {
"start": startTime,
"end": endTime,
sensor1.uuid: 2,
sensor2.uuid: 3
}
def test_should_forward_snapshot_once_data_complete():
manager = Manager()
manager.on_data = MagicMock()
snapper = Snapper(manager)
sensor1 = Sensor()
snapper.add_sensor(sensor1)
sensor1.publish(2, 1)
sensor2 = Sensor()
snapper.add_sensor(sensor2)
sensor2.publish(3, 1)
startTime = snapper.windowStart
endTime = snapper.windowEnd
# Force a snapshot publish by pushing value with time past window
sensor2.publish(3, 11)
manager.on_data.assert_called_with({
"start": startTime,
"end": endTime,
sensor1.uuid: 2,
sensor2.uuid: 3
})
class Manager:
def __init__(self):
self.sensors = []
self.route_map = {}
self.reverse_route_map = {}
self.variables = []
self.snapper = Snapper(self)
self.matrix = AssociationMatrix()
def set_window_size(self, new_window_size):
self.snapper.set_window_size(new_window_size)
def add_sensor(self, sensor):
"""
This function adds a new sensor to the snapper module and generates
a corresponding variable object with mappings.
:return:
"""
self.sensors.append(sensor)
self.snapper.add_sensor(sensor)
newVariable = Variable()
for var in self.variables:
relationship = SpearframeRelationship(newVariable, var)
self.matrix.add_relationship(relationship)
self.variables.append(newVariable)
self.route_map[sensor.uuid] = newVariable
self.reverse_route_map[str(newVariable.uuid)] = sensor.uuid
def remove_sensor(self, sensor):
"""
This function removes a new sensor to the snapper module, removes
variable from manager, and removes the corresponding relationships
from the matrix.
:return:
"""
variable = self.route_map[sensor.uuid]
self.route_map.pop(sensor.uuid)
self.reverse_route_map.pop(str(variable.uuid))
for var in self.variables:
if var == variable:
pass
else:
relationship = SpearframeRelationship(variable, var)
self.matrix.remove_relationship(relationship)
self.snapper.remove_sensor(sensor)
self.sensors.remove(sensor)
self.variables.remove(variable)
def get_matrix(self):
"""
Returns the underlying matrix on demand.
:return:
"""
return self.matrix
def get_value_matrix(self):
"""
Returns the underlying matrix on demand.
:return:
"""
return self.matrix.get_value_matrix()
def get_relationships_by_value_range(self, minvalue, maxvalue):
"""
Returns the underlying matrix on demand.
:return:
"""
return self.matrix.get_relationships_by_value_range(minvalue, maxvalue)
def on_data(self, snapshot):
"""
Routes all data from incoming snapshot to the appropriate variables.
:param snapshot:
:return:
"""
start = snapshot.pop("start")
stop = snapshot.pop("end")
for sensorID in snapshot:
variable = self.route_map[sensorID]
value = snapshot[sensorID]
variable.on_data(value, start, stop)
def get_relationship_from_sensors(self, sensor1, sensor2):
return self.matrix.get_relationship_from_sensors(sensor1, sensor2)
def get_all_relationships(self):
return self.matrix.relationships
def test_should_have_sensors_field_as_empty_list_on_init():
snapper = Snapper()
assert snapper.sensors == []
def test_should_store_sensor():
snapper = Snapper()
sensor = Sensor()
snapper.add_sensor(sensor)
assert snapper.sensors == [sensor]