def test_keep_hg_name(self): _input = Hypergraph() _input.name = "KEEP_ME" AddNodeToEdge(_input, 0, 0) AddNodeToEdge(_input, 0, 1) AddNodeToEdge(_input, 1, 1) actual_hg, removed_list = RemoveRandomConnections(_input, 0) self.assertEqual(actual_hg.name, "KEEP_ME")
def test_typical(self): _input = Hypergraph() AddNodeToEdge(_input, 0, 1) AddNodeToEdge(_input, 1, 1) AddNodeToEdge(_input, 1, 2) _input.name = "KEEP ME" node_map = {0: 100, 1: 200} edge_map = {1: 50, 2: 150} actual = Relabel(_input, node_map, edge_map) expected = Hypergraph() expected.name = "KEEP ME" AddNodeToEdge(expected, 100, 50) AddNodeToEdge(expected, 200, 50) AddNodeToEdge(expected, 200, 150) self.assertEqual(actual, expected)
def test_compress_range(self): original = Hypergraph() AddNodeToEdge(original, 100, 50) AddNodeToEdge(original, 200, 50) AddNodeToEdge(original, 200, 150) original.name = "KEEP ME" compressed, node_map, edge_map = CompressRange(original) restored = Relabel(compressed, node_map, edge_map) SparseArrayEquals(self, ToCsrMatrix(original), ToCsrMatrix(restored)) self.assertEqual(len(compressed.node), max(compressed.node) + 1) self.assertEqual(len(compressed.node), len(original.node)) self.assertEqual(len(compressed.edge), max(compressed.edge) + 1) self.assertEqual(len(compressed.edge), len(original.edge)) self.assertEqual(compressed.name, "KEEP ME")