def test_retrieval(self):
fusion_graph = FusionGraph(self.relations2)
self.assertEqual(fusion_graph.get_object_type('Type 1'), self.t1)
self.assertEqual(list(fusion_graph.get_relations(self.t1, self.t2)), self.relations2[:2])
self.assertEqual(fusion_graph[self.t1][self.t2], self.relations2[:2])
out_degree1 = len(list(fusion_graph.out_relations(self.t4)))
out_degree2 = sum(len(rels) for rels in fusion_graph[self.t4].values())
self.assertEqual(out_degree1, out_degree2)
def test_retrieval(self):
fusion_graph = FusionGraph(self.relations2)
self.assertEqual(fusion_graph.get_object_type('Type 1'), self.t1)
self.assertEqual(list(fusion_graph.get_relations(self.t1, self.t2)),
self.relations2[:2])
self.assertEqual(fusion_graph[self.t1][self.t2], self.relations2[:2])
out_degree1 = len(list(fusion_graph.out_relations(self.t4)))
out_degree2 = sum(len(rels) for rels in fusion_graph[self.t4].values())
self.assertEqual(out_degree1, out_degree2)
