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_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))
