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 <= 2, 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
        ]