def test_load_all_assistants_loads_proper_structure(self): YamlAssistantLoader.load_all_assistants(roles=['creator']) assert len(YamlAssistantLoader._assistants) == 1 assert len(YamlAssistantLoader._assistants['creator']) == 2 # ass is a really nice variable name, isn't it? a1 = YamlAssistantLoader._assistants['creator'][0] a2 = YamlAssistantLoader._assistants['creator'][1] if a1.name == 'c': c, f = a1, a2 else: c, f = a2, a1 assert len(c.get_subassistants()) == 2 assert len(f.get_subassistants()) == 1
def test_get_all_classes_sets_get_subassistants_properly(self): clss = YamlAssistantLoader.get_all_classes() for kls in clss: if kls.name == "c": assert set(map(lambda x: x.name, kls().get_subassistants())) == set(["d", "e"]) else: assert kls().get_subassistants() == []
def test_get_all_classes_sets_get_subassistants_properly(self): clss = YamlAssistantLoader.get_all_classes() for kls in clss: if kls.name == 'c': assert set(map(lambda x: x.name, kls().get_subassistants())) == set(['d', 'e']) else: assert kls().get_subassistants() == []
def create_or_refresh_cache(self, roles=settings.ASSISTANT_ROLES, assistants='assistants'): for role in roles: dirs = [ os.path.join(d, assistants, role) for d in settings.DATA_DIRECTORIES ] fh = YamlAssistantLoader.get_assistants_file_hierarchy(dirs) self.cch.refresh_role(role, fh)
def create_or_refresh_cache(self, roles=settings.ASSISTANT_ROLES, assistants='assistants'): for role in roles: dirs =[os.path.join(d, assistants, role) for d in settings.DATA_DIRECTORIES] fh = YamlAssistantLoader.get_assistants_file_hierarchy(dirs) self.cch.refresh_role(role, fh)
def test_get_top_level_assistants(self): clss = YamlAssistantLoader.get_top_level_assistants() assert set(['c', 'f']) == set(map(lambda x: x.name, clss))
def test_get_all_classes_loads_all(self): clss = YamlAssistantLoader.get_all_classes() assert len(clss) == 4 assert set(['c', 'd', 'e', 'f']) == set(map(lambda x: x.name, clss))
def test_get_top_level_assistants(self): ass = YamlAssistantLoader.get_top_level_assistants() assert set(['c', 'f']) == set(map(lambda x: x.name, ass))
def test_get_top_level_assistants(self): ass = YamlAssistantLoader.get_assistants( superassistants=[CreatorAssistant]) assert set(['c', 'f']) == set(map(lambda x: x.name, ass))
def test_get_top_level_assistants(self): clss = YamlAssistantLoader.get_top_level_assistants() assert set(["c", "f"]) == set(map(lambda x: x.name, clss))
def test_get_all_classes_loads_all(self): clss = YamlAssistantLoader.get_all_classes() assert len(clss) == 4 assert set(["c", "d", "e", "f"]) == set(map(lambda x: x.name, clss))