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