Mercurial > repos > guerler > springsuite
diff planemo/lib/python3.7/site-packages/networkx/tests/test_relabel.py @ 1:56ad4e20f292 draft
"planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"
author | guerler |
---|---|
date | Fri, 31 Jul 2020 00:32:28 -0400 (2020-07-31) |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/planemo/lib/python3.7/site-packages/networkx/tests/test_relabel.py Fri Jul 31 00:32:28 2020 -0400 @@ -0,0 +1,184 @@ +#!/usr/bin/env python +import pytest +import networkx as nx +from networkx.algorithms.operators import * +from networkx.generators.classic import empty_graph +from networkx.testing import assert_nodes_equal, assert_edges_equal + + +class TestRelabel(): + def test_convert_node_labels_to_integers(self): + # test that empty graph converts fine for all options + G = empty_graph() + H = nx.convert_node_labels_to_integers(G, 100) + assert list(H.nodes()) == [] + assert list(H.edges()) == [] + + for opt in ["default", "sorted", "increasing degree", "decreasing degree"]: + G = empty_graph() + H = nx.convert_node_labels_to_integers(G, 100, ordering=opt) + assert list(H.nodes()) == [] + assert list(H.edges()) == [] + + G = empty_graph() + G.add_edges_from([('A', 'B'), ('A', 'C'), ('B', 'C'), ('C', 'D')]) + H = nx.convert_node_labels_to_integers(G) + degH = (d for n, d in H.degree()) + degG = (d for n, d in G.degree()) + assert sorted(degH) == sorted(degG) + + H = nx.convert_node_labels_to_integers(G, 1000) + degH = (d for n, d in H.degree()) + degG = (d for n, d in G.degree()) + assert sorted(degH) == sorted(degG) + assert_nodes_equal(H.nodes(), [1000, 1001, 1002, 1003]) + + H = nx.convert_node_labels_to_integers(G, ordering="increasing degree") + degH = (d for n, d in H.degree()) + degG = (d for n, d in G.degree()) + assert sorted(degH) == sorted(degG) + assert H.degree(0) == 1 + assert H.degree(1) == 2 + assert H.degree(2) == 2 + assert H.degree(3) == 3 + + H = nx.convert_node_labels_to_integers(G, ordering="decreasing degree") + degH = (d for n, d in H.degree()) + degG = (d for n, d in G.degree()) + assert sorted(degH) == sorted(degG) + assert H.degree(0) == 3 + assert H.degree(1) == 2 + assert H.degree(2) == 2 + assert H.degree(3) == 1 + + H = nx.convert_node_labels_to_integers(G, ordering="increasing degree", + label_attribute='label') + degH = (d for n, d in H.degree()) + degG = (d for n, d in G.degree()) + assert sorted(degH) == sorted(degG) + assert H.degree(0) == 1 + assert H.degree(1) == 2 + assert H.degree(2) == 2 + assert H.degree(3) == 3 + + # check mapping + assert H.nodes[3]['label'] == 'C' + assert H.nodes[0]['label'] == 'D' + assert H.nodes[1]['label'] == 'A' or H.nodes[2]['label'] == 'A' + assert H.nodes[1]['label'] == 'B' or H.nodes[2]['label'] == 'B' + + def test_convert_to_integers2(self): + G = empty_graph() + G.add_edges_from([('C', 'D'), ('A', 'B'), ('A', 'C'), ('B', 'C')]) + H = nx.convert_node_labels_to_integers(G, ordering="sorted") + degH = (d for n, d in H.degree()) + degG = (d for n, d in G.degree()) + assert sorted(degH) == sorted(degG) + + H = nx.convert_node_labels_to_integers(G, ordering="sorted", + label_attribute='label') + assert H.nodes[0]['label'] == 'A' + assert H.nodes[1]['label'] == 'B' + assert H.nodes[2]['label'] == 'C' + assert H.nodes[3]['label'] == 'D' + + def test_convert_to_integers_raise(self): + with pytest.raises(nx.NetworkXError): + G = nx.Graph() + H = nx.convert_node_labels_to_integers(G, ordering="increasing age") + + def test_relabel_nodes_copy(self): + G = nx.empty_graph() + G.add_edges_from([('A', 'B'), ('A', 'C'), ('B', 'C'), ('C', 'D')]) + mapping = {'A': 'aardvark', 'B': 'bear', 'C': 'cat', 'D': 'dog'} + H = nx.relabel_nodes(G, mapping) + assert_nodes_equal(H.nodes(), ['aardvark', 'bear', 'cat', 'dog']) + + def test_relabel_nodes_function(self): + G = nx.empty_graph() + G.add_edges_from([('A', 'B'), ('A', 'C'), ('B', 'C'), ('C', 'D')]) + # function mapping no longer encouraged but works + + def mapping(n): + return ord(n) + H = nx.relabel_nodes(G, mapping) + assert_nodes_equal(H.nodes(), [65, 66, 67, 68]) + + def test_relabel_nodes_graph(self): + G = nx.Graph([('A', 'B'), ('A', 'C'), ('B', 'C'), ('C', 'D')]) + mapping = {'A': 'aardvark', 'B': 'bear', 'C': 'cat', 'D': 'dog'} + H = nx.relabel_nodes(G, mapping) + assert_nodes_equal(H.nodes(), ['aardvark', 'bear', 'cat', 'dog']) + + def test_relabel_nodes_orderedgraph(self): + G = nx.OrderedGraph() + G.add_nodes_from([1, 2, 3]) + G.add_edges_from([(1, 3), (2, 3)]) + mapping = {1: 'a', 2: 'b', 3: 'c'} + H = nx.relabel_nodes(G, mapping) + assert list(H.nodes) == ['a', 'b', 'c'] + + def test_relabel_nodes_digraph(self): + G = nx.DiGraph([('A', 'B'), ('A', 'C'), ('B', 'C'), ('C', 'D')]) + mapping = {'A': 'aardvark', 'B': 'bear', 'C': 'cat', 'D': 'dog'} + H = nx.relabel_nodes(G, mapping, copy=False) + assert_nodes_equal(H.nodes(), ['aardvark', 'bear', 'cat', 'dog']) + + def test_relabel_nodes_multigraph(self): + G = nx.MultiGraph([('a', 'b'), ('a', 'b')]) + mapping = {'a': 'aardvark', 'b': 'bear'} + G = nx.relabel_nodes(G, mapping, copy=False) + assert_nodes_equal(G.nodes(), ['aardvark', 'bear']) + assert_edges_equal(G.edges(), [('aardvark', 'bear'), ('aardvark', 'bear')]) + + def test_relabel_nodes_multidigraph(self): + G = nx.MultiDiGraph([('a', 'b'), ('a', 'b')]) + mapping = {'a': 'aardvark', 'b': 'bear'} + G = nx.relabel_nodes(G, mapping, copy=False) + assert_nodes_equal(G.nodes(), ['aardvark', 'bear']) + assert_edges_equal(G.edges(), [('aardvark', 'bear'), ('aardvark', 'bear')]) + + def test_relabel_isolated_nodes_to_same(self): + G = nx.Graph() + G.add_nodes_from(range(4)) + mapping = {1: 1} + H = nx.relabel_nodes(G, mapping, copy=False) + assert_nodes_equal(H.nodes(), list(range(4))) + + def test_relabel_nodes_missing(self): + with pytest.raises(KeyError): + G = nx.Graph([('A', 'B'), ('A', 'C'), ('B', 'C'), ('C', 'D')]) + mapping = {0: 'aardvark'} + G = nx.relabel_nodes(G, mapping, copy=False) + + def test_relabel_copy_name(self): + G = nx.Graph() + H = nx.relabel_nodes(G, {}, copy=True) + assert H.graph == G.graph + H = nx.relabel_nodes(G, {}, copy=False) + assert H.graph == G.graph + G.name = "first" + H = nx.relabel_nodes(G, {}, copy=True) + assert H.graph == G.graph + H = nx.relabel_nodes(G, {}, copy=False) + assert H.graph == G.graph + + def test_relabel_toposort(self): + K5 = nx.complete_graph(4) + G = nx.complete_graph(4) + G = nx.relabel_nodes(G, dict([(i, i + 1) for i in range(4)]), copy=False) + nx.is_isomorphic(K5, G) + G = nx.complete_graph(4) + G = nx.relabel_nodes(G, dict([(i, i - 1) for i in range(4)]), copy=False) + nx.is_isomorphic(K5, G) + + def test_relabel_selfloop(self): + G = nx.DiGraph([(1, 1), (1, 2), (2, 3)]) + G = nx.relabel_nodes(G, {1: 'One', 2: 'Two', 3: 'Three'}, copy=False) + assert_nodes_equal(G.nodes(), ['One', 'Three', 'Two']) + G = nx.MultiDiGraph([(1, 1), (1, 2), (2, 3)]) + G = nx.relabel_nodes(G, {1: 'One', 2: 'Two', 3: 'Three'}, copy=False) + assert_nodes_equal(G.nodes(), ['One', 'Three', 'Two']) + G = nx.MultiDiGraph([(1, 1)]) + G = nx.relabel_nodes(G, {1: 0}, copy=False) + assert_nodes_equal(G.nodes(), [0])