def test_set_max_trials_with_exit_policy(self) -> None: """Test overwriting max_trials in Tuning config.""" tuning = Tuning({ "exit_policy": { "max_trials": 60, }, }, ) self.assertIsNotNone(tuning.exit_policy) self.assertEqual(tuning.exit_policy.max_trials, 60) tuning.set_max_trials(10) self.assertIsNotNone(tuning.exit_policy) self.assertEqual(tuning.exit_policy.max_trials, 10)
def test_set_max_trials_invalid_string(self) -> None: """Test overwriting max_trials in Tuning config.""" tuning = Tuning() with self.assertRaises(ClientErrorException): tuning.set_max_trials("abc")
def test_set_max_trials_negative(self) -> None: """Test overwriting max_trials in Tuning config.""" tuning = Tuning() with self.assertRaises(ClientErrorException): tuning.set_max_trials(-1)
def test_set_max_trials_from_string(self) -> None: """Test overwriting max_trials in Tuning config.""" tuning = Tuning() tuning.set_max_trials("10") self.assertIsNotNone(tuning.exit_policy) self.assertEqual(tuning.exit_policy.max_trials, 10)
def test_set_max_trials(self) -> None: """Test setting max_trials in Tuning config.""" tuning = Tuning() tuning.set_max_trials(10) self.assertIsNotNone(tuning.exit_policy) self.assertEqual(tuning.exit_policy.max_trials, 10)