def test_get_default_pstate_valid(self): p1 = PowerState(0, PowerStateType.SLEEP, 10, 10) p2 = PowerState(1, PowerStateType.COMPUTATION, 10, 100) p3 = PowerState(2, PowerStateType.COMPUTATION, 1000, 10000) p4 = PowerState(3, PowerStateType.SWITCHING_OFF, 50, 50) p5 = PowerState(4, PowerStateType.SWITCHING_ON, 25, 25) h = Host(0, "n", pstates=[p1, p3, p2, p4, p5]) assert h.get_default_pstate() == p2
def test_get_default_pstate_not_defined_must_raise(self): h = Host(0, "n", pstates=[]) with pytest.raises(RuntimeError) as excinfo: h.get_default_pstate() assert "undefined" in str(excinfo.value)