test_pelt_sim.py

from hypothesis import given
from hypothesis.strategies import integers
import unittest

from pelt import *

class TestSimulator(unittest.TestCase):

    @given(x=integers(min_value=1), y=integers(min_value=1))
    def test_range_below_max(self, x, y):
        period = max(x, y)
        run = min(x, y)

        t = Task(period, run)
        s = Simulator()
        r = s.stable_range(t)
        self.assertLessEqual(r.max_value, s._signal_max)

    @given(x=integers(min_value=1), y=integers(min_value=1))
    def test_signal_within_range(self, x, y):
        period = max(x, y)
        run = min(x, y)

        t = Task(period, run)
        s = Simulator(init_util=0)
        stable_range = s.stable_range(t)
        signal = s.signal(t)
        signal_max = signal.max()['util_avg']
        self.assertLessEqual(signal_max, s._signal_max)

if __name__ == "__main__":
    unittest.main()