def test_build_entity_version_by_entity_id_parent(self): all_current_entities_version = entity_version.find_all_current_entities_version( ) result = entity_version._build_entity_version_by_entity_id( all_current_entities_version) expected_keys = [v.entity_id for v in all_current_entities_version] self.assertListEqual(list(sorted(result.keys())), sorted(expected_keys))
def test_build_direct_children_by_entity_version_id(self): entity_version_by_entity_id = entity_version._build_entity_version_by_entity_id( entity_version.find_all_current_entities_version()) result = entity_version._build_direct_children_by_entity_version_id(entity_version_by_entity_id) count_entities_version_with_children = 4 self.assertEqual(len(result.keys()), count_entities_version_with_children) root_direct_children = [self.SC, self.LOCI] self.assertEqual(set(result[self.root.id]), set(root_direct_children)) sc_direct_children = [self.MATH, self.PHYS] self.assertEqual(set(result[self.SC.id]), set(sc_direct_children)) self.assertNotIn(self.MATH.id, result) # No children for MATH
def test_build_all_children_by_entity_version_id(self): all_current_entites_versions = entity_version.find_all_current_entities_version( ) entity_version_by_entity_id = entity_version._build_entity_version_by_entity_id( all_current_entites_versions) direct_children_by_entity_version_id = entity_version \ ._build_direct_children_by_entity_version_id(entity_version_by_entity_id) result = entity_version._build_all_children_by_entity_version_id( direct_children_by_entity_version_id) count_entities_version_with_children = 4 self.assertEqual(len(result.keys()), count_entities_version_with_children) root_all_children = [ self.SC, self.LOCI, self.MATH, self.PHYS, self.URBA, self.BARC ] self.assertEqual(set(result[self.root.id]), set(root_all_children)) sc_all_children = [self.MATH, self.PHYS] self.assertEqual(set(result[self.SC.id]), set(sc_all_children)) self.assertNotIn(self.MATH.id, result.keys())