def test_extract_timestamps_multi(self):
cycles = []
for i in range(3):
cycle = opqbox3_pb2.Cycle()
cycle.timestamp_ms = i
cycles.append(cycle)
self.assertEqual(trigger_plugin.extract_timestamps(cycles), (0, 2))
def test_cycles_to_data(self):
cycles = []
for i in range(0, 100, 10):
cycle = opqbox3_pb2.Cycle()
cycle.datapoints.extend(list(range(i, i + 10)))
cycles.append(cycle)
self.assertEqual(cycles_to_data(cycles), list(range(0, 100)))
def deserialize_makai_cycle(cycle_bytes: bytes) -> opqbox3_pb2.Cycle:
"""
Deserializes a makai cycle.
:param cycle_bytes: Serialized makai cycle.
:return: Deserialized makai cycle.
"""
cycle = opqbox3_pb2.Cycle()
cycle.ParseFromString(cycle_bytes)
return cycle
def test_extract_timestamps_single(self):
cycle = opqbox3_pb2.Cycle()
cycle.timestamp_ms = 1
self.assertEqual(trigger_plugin.extract_timestamps([cycle]), (1, 1))