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)
