def test_inspection(self):
fusion_graph = FusionGraph(self.relations2)
self.assertEqual(set(fusion_graph.in_relations(self.t1)),
{self.relations2[6]})
self.assertEqual(set(fusion_graph.out_relations(self.t1)),
set(self.relations2[:2]))
out_nbs = {self.relations2[4], self.relations2[7], self.relations2[8]}
self.assertEqual(set(fusion_graph.out_relations(self.t4)), out_nbs)
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)
def test_inspection(self):
fusion_graph = FusionGraph(self.relations2)
self.assertEqual(set(fusion_graph.in_relations(self.t1)), {self.relations2[6]})
self.assertEqual(set(fusion_graph.out_relations(self.t1)), set(self.relations2[:2]))
out_nbs = {self.relations2[4], self.relations2[7], self.relations2[8]}
self.assertEqual(set(fusion_graph.out_relations(self.t4)), out_nbs)
