1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
from aergia.aergia import Aergia
import unittest
class AergiaUnitTestCase(unittest.TestCase):
interval = 0.01
Aergia = Aergia(interval)
def setUp(self):
self.Aergia.clear()
def assertFuncContains(self, func_name, samples_expected, samples):
samples_actual = self.extract_values_by_func(samples,
func_name)
self.assertTrue(len(samples_expected) == len(samples_actual),
f'{samples_expected} (expected) not length of '
f'{samples_actual} (actual)')
s1 = sorted(samples_expected)
s2 = sorted(samples_actual)
for v1, v2 in zip(s1, s2):
self.assertRoughlyEqual(v1, v2)
def assertRoughlyEqual(self, v1, v2):
a = abs(v1 - v2)
self.assertTrue(a <= 1, f'{v1} (expected) not roughly {v2} (actual)')
def expected_samples(self, total_seconds):
return (total_seconds / self.interval)
def extract_values_by_func(self, samples, func_name):
return [
value for key, value in samples.items()
if key.func == func_name
]
|
