def test_set_timeout_with_exit_policy(self) -> None: """Test overwriting timeout in Tuning config.""" tuning = Tuning({ "exit_policy": { "timeout": 60, }, }, ) self.assertIsNotNone(tuning.exit_policy) self.assertEqual(tuning.exit_policy.timeout, 60) tuning.set_timeout(10) self.assertIsNotNone(tuning.exit_policy) self.assertEqual(tuning.exit_policy.timeout, 10)
def test_set_timeout_invalid_string(self) -> None: """Test overwriting timeout in Tuning config.""" tuning = Tuning() with self.assertRaises(ClientErrorException): tuning.set_timeout("abc")
def test_set_timeout_negative(self) -> None: """Test overwriting timeout in Tuning config.""" tuning = Tuning() with self.assertRaises(ClientErrorException): tuning.set_timeout(-1)
def test_set_timeout_from_string(self) -> None: """Test overwriting timeout in Tuning config.""" tuning = Tuning() tuning.set_timeout("10") self.assertIsNotNone(tuning.exit_policy) self.assertEqual(tuning.exit_policy.timeout, 10)
def test_set_timeout(self) -> None: """Test setting timeout in Tuning config.""" tuning = Tuning() tuning.set_timeout(10) self.assertIsNotNone(tuning.exit_policy) self.assertEqual(tuning.exit_policy.timeout, 10)