def test_get_names_by_object_type(self):
rnds = np.random.RandomState(0)
X = rnds.rand(10, 10)
t1_names = list('ABCDEFGHIJ')
t2_names = list('KLMNOPQRST')
rel = Relation(X,
name='Test',
row_type=self.t1,
row_names=t1_names,
col_type=self.t2,
col_names=t2_names)
rel2 = Relation(X,
name='Test2',
row_type=self.t2,
row_names=t2_names,
col_type=self.t3)
fusion_graph = FusionGraph()
fusion_graph.add_relation(rel)
fusion_graph.add_relation(rel2)
self.assertEqual(fusion_graph.get_names(self.t1), t1_names)
self.assertEqual(fusion_graph.get_names(self.t2), t2_names)
t3_names = fusion_graph.get_names(self.t3)
self.assertEqual(len(t3_names), 10)
def test_get_names_by_object_type(self):
rnds = np.random.RandomState(0)
X = rnds.rand(10, 10)
t1_names = list('ABCDEFGHIJ')
t2_names = list('KLMNOPQRST')
rel = Relation(X, name='Test',
row_type=self.t1, row_names=t1_names,
col_type=self.t2, col_names=t2_names)
rel2 = Relation(X, name='Test2',
row_type=self.t2, row_names=t2_names,
col_type=self.t3)
fusion_graph = FusionGraph()
fusion_graph.add_relation(rel)
fusion_graph.add_relation(rel2)
self.assertEqual(fusion_graph.get_names(self.t1), t1_names)
self.assertEqual(fusion_graph.get_names(self.t2), t2_names)
t3_names = fusion_graph.get_names(self.t3)
self.assertEqual(len(t3_names), 10)
