def test_find_cores2(self): core = nx.find_cores(self.H) nodes_by_core = [ sorted([n for n in core if core[n] == val]) for val in range(3) ] assert nodes_equal(nodes_by_core[0], [0]) assert nodes_equal(nodes_by_core[1], [1, 3]) assert nodes_equal(nodes_by_core[2], [2, 4, 5, 6])
def test_find_cores(self): core = nx.find_cores(self.G) nodes_by_core = [ sorted([n for n in core if core[n] == val]) for val in range(4) ] assert nodes_equal(nodes_by_core[0], [21]) assert nodes_equal(nodes_by_core[1], [17, 18, 19, 20]) assert nodes_equal(nodes_by_core[2], [9, 10, 11, 12, 13, 14, 15, 16]) assert nodes_equal(nodes_by_core[3], [1, 2, 3, 4, 5, 6, 7, 8])
def test_trivial(self): """Empty graph""" G = nx.Graph() assert nx.find_cores(G) == {}