class LinearLearnerTimeTest(unittest.TestCase): def setUp(self): _ = mock.patch.object(LinearLearner, 'learn') mock_learn = _.start() self.addCleanup(_.stop) _ = mock.patch.object(LinearLearner, 'learned_maximum_load', new_callable = mock.PropertyMock) self.mock_max_load = _.start() self.addCleanup(_.stop) _ = mock.patch('pmort.learners.linear.os.getloadavg') self.mock_load = _.start() self.addCleanup(_.stop) _ = mock.patch('pmort.learners.linear.PARAMETERS') mock_parameters = _.start() self.addCleanup(_.stop) mock_parameters.__getitem__.side_effect = lambda _: { 'learner.minimum_interval': 1, 'learner.maximum_interval': 600, }[_] self.l = LinearLearner() def test_linear_times(self): '''Linear Times''' loads = ( (0.0, 1.0, 600), (0.5, 1.0, 300.5), (1.0, 1.0, 1), (1.0, 0.5, 1), ) for load in loads: self.mock_load.return_value = (load[0],) self.mock_max_load.return_value = load[1] self.assertEqual(load[2], self.l.time())
def setUp(self): _ = mock.patch.object(LinearLearner, 'learn') mock_learn = _.start() self.addCleanup(_.stop) _ = mock.patch.object(LinearLearner, 'learned_maximum_load', new_callable = mock.PropertyMock) self.mock_max_load = _.start() self.addCleanup(_.stop) _ = mock.patch('pmort.learners.linear.os.getloadavg') self.mock_load = _.start() self.addCleanup(_.stop) _ = mock.patch('pmort.learners.linear.PARAMETERS') mock_parameters = _.start() self.addCleanup(_.stop) mock_parameters.__getitem__.side_effect = lambda _: { 'learner.minimum_interval': 1, 'learner.maximum_interval': 600, }[_] self.l = LinearLearner()