def test_construct_model_names_only_some_names():
params = pd.DataFrame()
info1 = {"model_name": "small_1", "foo": "bar"}
info2 = {"foo": "bar2"}
results_with_only_some_names = [OPT_RES(params, info1), OPT_RES(params, info2)]
with pytest.raises(AssertionError):
test_module._construct_model_names(results_with_only_some_names)
def test_construct_model_names_unique_names():
params = pd.DataFrame()
info1 = {"model_name": "small_1", "foo": "bar"}
info2 = {"model_name": "small_2", "foo": "bar2"}
unique_name_results = [OPT_RES(params, info1), OPT_RES(params, info2)]
res = test_module._construct_model_names(results=unique_name_results)
expected = ["small_1", "small_2"]
assert res == expected
def test_construct_model_names_no_names():
params = pd.DataFrame()
info1 = {"model_class": "small", "foo": "bar"}
info2 = {"model_class": "large", "foo": "bar2"}
no_name_results = [OPT_RES(params, info1), OPT_RES(params, info2)]
res = test_module._construct_model_names(results=no_name_results)
expected = ["0", "1"]
assert res == expected
