def test_dist_weights(self): id2linkpoints, id2attr, link2id = lincs.edgepoints_from_network(self.G) w, id2link = lincs.dist_weights(self.G, id2linkpoints, link2id, 500) self.assertEqual(w.neighbors[0],[1,154,153,155]) self.assertEqual(id2link[0], self.test_link)
def test_dist_weights(self): id2linkpoints, id2attr, link2id = lincs.edgepoints_from_network(self.G) w, id2link = lincs.dist_weights(self.G, id2linkpoints, link2id, 500) self.assertEqual(w.neighbors[0], [1, 154, 153, 155]) self.assertEqual(id2link[0], self.test_link)
def test_edgepoints_from_network(self): id2linkpoints, id2attr, link2id = lincs.edgepoints_from_network(self.G) link = id2linkpoints[0] self.assertEqual(link[:2], self.test_link) self.assertEqual(link2id[link[:2]], 0)