def test_metric_with_eval_at(self, metric, eval_at_param, expected):
# type: (str, Dict[str, Union[int, List[int]]], str) -> None
params = {"metric": metric} # type: Dict[str, Union[str, int, List[int]]]
params.update(eval_at_param)
tuner = BaseTuner(lgbm_params=params)
assert tuner._metric_with_eval_at(metric) == expected
def test_metric_with_eval_at_error(self):
# type: () -> None
tuner = BaseTuner(lgbm_params={"metric": "ndcg", "eval_at": "1"})
with pytest.raises(ValueError):
tuner._metric_with_eval_at("ndcg")
def test_metric_with_eval_at_error(self):
# type: () -> None
tuner = BaseTuner(lgbm_params={'metric': 'ndcg', 'eval_at': '1'})
with pytest.raises(ValueError):
tuner._metric_with_eval_at('ndcg')
