def test_too_high_a_timeout_raises_errors(self): timeout_est = under_test.TimeoutEstimate( max_test_runtime=5, expected_task_runtime=under_test.MAX_EXPECTED_TIMEOUT) with self.assertRaises(ValueError): timeout_est.generate_timeout_cmd(is_patch=True, repeat_factor=1)
def generate_mock_suites(count): suites = [] for idx in range(count): suite = MagicMock() suite.name = "suite {0}".format(idx) suite.max_runtime = 5.28 suite.get_runtime = lambda: 100.874 suite.get_timeout_estimate.return_value = under_test.TimeoutEstimate( max_test_runtime=5.28, expected_task_runtime=100.874) suites.append(suite) return suites