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")
