def test_transform_buffer_sums_currents(self):
sample_aggregator = SampleAggregator()
sample_aggregator._transform_buffer([
HvpmReading([1.41421356237, 0, 0, 0, 0], 0.01),
HvpmReading([2.71828182846, 0, 0, 0, 0], 0.99),
HvpmReading([3.14159265359, 0, 0, 0, 0], 1.00),
])
self.assertEqual(sample_aggregator.num_samples, 3)
self.assertAlmostEqual(sample_aggregator.sum_currents, 7.27408804442)
def test_transform_buffer_respects_start_after_seconds_flag(self):
sample_aggregator = SampleAggregator(start_after_seconds=1.0)
sample_aggregator._transform_buffer([
HvpmReading([1.41421356237, 0, 0, 0, 0], 0.00),
HvpmReading([2.71828182846, 0, 0, 0, 0], 0.99),
HvpmReading([3.14159265359, 0, 0, 0, 0], 1.00),
])
self.assertEqual(sample_aggregator.num_samples, 1)
self.assertEqual(sample_aggregator.sum_currents, 3.14159265359)
def test_transform_stores_unused_values_in_leftovers(self):
downsampler = DownSampler(3)
buffer = [
HvpmReading([2, 0, 0, 0, 0], .01),
HvpmReading([4, 0, 0, 0, 0], .03),
HvpmReading([6, 0, 0, 0, 0], .05),
HvpmReading([8, 0, 0, 0, 0], .07),
HvpmReading([10, 0, 0, 0, 0], .09),
]
downsampler._transform_buffer(buffer)
self.assertEqual(len(downsampler._leftovers), 2)
self.assertIn(buffer[-2], downsampler._leftovers)
self.assertIn(buffer[-1], downsampler._leftovers)
def test_transform_buffer_outputs_correct_format(self, open_mock):
tee = Tee('foo')
tee.on_begin()
expected_output = [
'0.010000000 1.414213562370\n', '0.020000000 2.718281828460\n',
'0.030000000 3.141592653590\n'
]
tee._transform_buffer([
HvpmReading([1.41421356237, 0, 0, 0, 0], 0.01),
HvpmReading([2.71828182846, 0, 0, 0, 0], 0.02),
HvpmReading([3.14159265359, 0, 0, 0, 0], 0.03),
])
for call, out in zip(open_mock().write.call_args_list,
expected_output):
self.assertEqual(call[ARGS][0], out)
def test_transform_buffer_outputs_correct_format(self, open_mock):
tee = PerfgateTee('foo')
tee.on_begin()
expected_output = [
'1596149635552503296,0.000223,4.193050\n',
'1596149635562476032,0.000212,4.193190\n',
'1596149635572549376,0.000225,4.193135\n',
]
tee._transform_buffer([
HvpmReading([0.000223, 0, 0, 4.193050, 0], 1596149635.552503296),
HvpmReading([0.000212, 0, 0, 4.193190, 0], 1596149635.562476032),
HvpmReading([0.000225, 0, 0, 4.193135, 0], 1596149635.572549376),
])
for call, out in zip(open_mock().write.call_args_list,
expected_output):
self.assertEqual(call[ARGS][0], out)
def test_transform_uses_leftovers_on_next_calculation(self):
downsampler = DownSampler(3)
starting_leftovers = [
HvpmReading([2, 0, 0, 0, 0], .01),
HvpmReading([4, 0, 0, 0, 0], .03),
]
downsampler._leftovers = starting_leftovers
buffer = [
HvpmReading([6, 0, 0, 0, 0], .05),
HvpmReading([8, 0, 0, 0, 0], .07),
HvpmReading([10, 0, 0, 0, 0], .09),
HvpmReading([12, 0, 0, 0, 0], .011)
]
values = downsampler._transform_buffer(buffer)
self.assertEqual(len(values), 2)
self.assertNotIn(starting_leftovers[0], downsampler._leftovers)
self.assertNotIn(starting_leftovers[1], downsampler._leftovers)
self.assertAlmostEqual(
values[0].main_current,
statistics.mean([
starting_leftovers[0].main_current,
starting_leftovers[1].main_current,
buffer[0].main_current,
]))
self.assertAlmostEqual(
values[1].main_current,
statistics.mean([
buffer[1].main_current,
buffer[2].main_current,
buffer[3].main_current,
]))
def test_transform_buffer_downsamples_without_leftovers(self):
downsampler = DownSampler(2)
buffer = [
HvpmReading([2, 0, 0, 0, 0], .01),
HvpmReading([4, 0, 0, 0, 0], .03),
HvpmReading([6, 0, 0, 0, 0], .05),
HvpmReading([8, 0, 0, 0, 0], .07),
HvpmReading([10, 0, 0, 0, 0], .09),
HvpmReading([12, 0, 0, 0, 0], .011),
]
values = downsampler._transform_buffer(buffer)
self.assertEqual(len(values), len(buffer) / 2)
for i, down_sample in enumerate(values):
self.assertAlmostEqual(
down_sample.main_current,
((buffer[2 * i] + buffer[2 * i + 1]) / 2).main_current)