def test_format_named_configs(): ingred = Ingredient('ingred') ex = Experiment(name='experiment', ingredients=[ingred]) @ingred.named_config def named_config1(): pass @ex.named_config def named_config2(): """named config with doc""" pass dict_config = dict(v=42) ingred.add_named_config('dict_config', dict_config) named_configs_text = _format_named_configs( OrderedDict(ex.gather_named_configs())) assert named_configs_text.startswith('Named Configurations (' + COLOR_DOC + 'doc' + ENDC + '):') assert 'named_config2' in named_configs_text assert '# named config with doc' in named_configs_text assert 'ingred.named_config1' in named_configs_text assert 'ingred.dict_config' in named_configs_text
def test_format_named_configs(): ingred = Ingredient("ingred") ex = Experiment(name="experiment", ingredients=[ingred]) @ingred.named_config def named_config1(): pass @ex.named_config def named_config2(): """named config with doc""" pass dict_config = dict(v=42) ingred.add_named_config("dict_config", dict_config) named_configs_text = _format_named_configs( OrderedDict(ex.gather_named_configs())) assert named_configs_text.startswith("Named Configurations (" + COLOR_DOC + "doc" + ENDC + "):") assert "named_config2" in named_configs_text assert "# named config with doc" in named_configs_text assert "ingred.named_config1" in named_configs_text assert "ingred.dict_config" in named_configs_text