diff options
author | Jarrod Millman <jarrod.millman@gmail.com> | 2021-05-26 08:59:54 -0700 |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-05-26 08:59:54 -0700 |
commit | 09b30ba69dd2e81a2728fd1259ca7b4d88470827 (patch) | |
tree | ec1c723f51473a8ff99fb0195783c607242bb90c /networkx/readwrite/tests/test_graphml.py | |
parent | 7771b28bad07f0a07ac08ef8c2c6a17191d36f54 (diff) | |
download | networkx-09b30ba69dd2e81a2728fd1259ca7b4d88470827.tar.gz |
Refactor testing utilities (#4829)
* Refactor testing utilities
Change `assert_edges_equal`, `assert_graphs_equal`, and `assert_nodes_equal`
to be more pytest-idiomatic.
For example, `assert_edges_equal` becomes the Boolean function `edges_equal`
and then the assert is done the testing file
(i.e., `assert edges_equal(edges1, edges2)`).
This also makes these utility functions useful in nontesting situations
where you want to compare edges, but not raise an exception based on the result.
* Move testing utility functions
* Use new testing utilities
* Deprecate assert_*_equal testing utilities
* Document node, edge, and graph equality helper functions
* text nits.
* Update networkx/tests/test_convert_pandas.py
Co-authored-by: Ross Barnowski <rossbar@berkeley.edu>
* Update networkx/readwrite/tests/test_sparse6.py
Co-authored-by: Ross Barnowski <rossbar@berkeley.edu>
* Update networkx/readwrite/tests/test_graph6.py
Co-authored-by: Ross Barnowski <rossbar@berkeley.edu>
* Update networkx/generators/tests/test_classic.py
Co-authored-by: Ross Barnowski <rossbar@berkeley.edu>
* Update networkx/algorithms/tree/tests/test_operations.py
Co-authored-by: Ross Barnowski <rossbar@berkeley.edu>
* Update networkx/algorithms/tree/tests/test_coding.py
Co-authored-by: Ross Barnowski <rossbar@berkeley.edu>
* Update networkx/algorithms/tests/test_dag.py
Co-authored-by: Ross Barnowski <rossbar@berkeley.edu>
* Update networkx/algorithms/minors/tests/test_contraction.py
Co-authored-by: Ross Barnowski <rossbar@berkeley.edu>
* add short equality description to docstring
* Suppress known warnings
Co-authored-by: Ross Barnowski <rossbar@berkeley.edu>
Co-authored-by: Dan Schult <dschult@colgate.edu>
Diffstat (limited to 'networkx/readwrite/tests/test_graphml.py')
-rw-r--r-- | networkx/readwrite/tests/test_graphml.py | 56 |
1 files changed, 28 insertions, 28 deletions
diff --git a/networkx/readwrite/tests/test_graphml.py b/networkx/readwrite/tests/test_graphml.py index 75f1a91b..fbbc1953 100644 --- a/networkx/readwrite/tests/test_graphml.py +++ b/networkx/readwrite/tests/test_graphml.py @@ -1,6 +1,6 @@ import pytest import networkx as nx -from networkx.testing.utils import assert_edges_equal, assert_nodes_equal +from networkx.utils import nodes_equal, edges_equal from networkx.readwrite.graphml import GraphMLWriter import io import tempfile @@ -293,51 +293,51 @@ class TestReadGraphML(BaseGraphML): def test_read_simple_undirected_graphml(self): G = self.simple_undirected_graph H = nx.read_graphml(self.simple_undirected_fh) - assert_nodes_equal(G.nodes(), H.nodes()) - assert_edges_equal(G.edges(), H.edges()) + assert nodes_equal(G.nodes(), H.nodes()) + assert edges_equal(G.edges(), H.edges()) self.simple_undirected_fh.seek(0) PG = nx.parse_graphml(self.simple_undirected_data) - assert_nodes_equal(G.nodes(), PG.nodes()) - assert_edges_equal(G.edges(), PG.edges()) + assert nodes_equal(G.nodes(), PG.nodes()) + assert edges_equal(G.edges(), PG.edges()) def test_read_undirected_multigraph_graphml(self): G = self.undirected_multigraph H = nx.read_graphml(self.undirected_multigraph_fh) - assert_nodes_equal(G.nodes(), H.nodes()) - assert_edges_equal(G.edges(), H.edges()) + assert nodes_equal(G.nodes(), H.nodes()) + assert edges_equal(G.edges(), H.edges()) self.undirected_multigraph_fh.seek(0) PG = nx.parse_graphml(self.undirected_multigraph_data) - assert_nodes_equal(G.nodes(), PG.nodes()) - assert_edges_equal(G.edges(), PG.edges()) + assert nodes_equal(G.nodes(), PG.nodes()) + assert edges_equal(G.edges(), PG.edges()) def test_read_undirected_multigraph_no_multiedge_graphml(self): G = self.undirected_multigraph_no_multiedge H = nx.read_graphml(self.undirected_multigraph_no_multiedge_fh) - assert_nodes_equal(G.nodes(), H.nodes()) - assert_edges_equal(G.edges(), H.edges()) + assert nodes_equal(G.nodes(), H.nodes()) + assert edges_equal(G.edges(), H.edges()) self.undirected_multigraph_no_multiedge_fh.seek(0) PG = nx.parse_graphml(self.undirected_multigraph_no_multiedge_data) - assert_nodes_equal(G.nodes(), PG.nodes()) - assert_edges_equal(G.edges(), PG.edges()) + assert nodes_equal(G.nodes(), PG.nodes()) + assert edges_equal(G.edges(), PG.edges()) def test_read_undirected_multigraph_only_ids_for_multiedges_graphml(self): G = self.multigraph_only_ids_for_multiedges H = nx.read_graphml(self.multigraph_only_ids_for_multiedges_fh) - assert_nodes_equal(G.nodes(), H.nodes()) - assert_edges_equal(G.edges(), H.edges()) + assert nodes_equal(G.nodes(), H.nodes()) + assert edges_equal(G.edges(), H.edges()) self.multigraph_only_ids_for_multiedges_fh.seek(0) PG = nx.parse_graphml(self.multigraph_only_ids_for_multiedges_data) - assert_nodes_equal(G.nodes(), PG.nodes()) - assert_edges_equal(G.edges(), PG.edges()) + assert nodes_equal(G.nodes(), PG.nodes()) + assert edges_equal(G.edges(), PG.edges()) def test_read_attribute_graphml(self): G = self.attribute_graph H = nx.read_graphml(self.attribute_fh) - assert_nodes_equal(G.nodes(True), sorted(H.nodes(data=True))) + assert nodes_equal(G.nodes(True), sorted(H.nodes(data=True))) ge = sorted(G.edges(data=True)) he = sorted(H.edges(data=True)) for a, b in zip(ge, he): @@ -494,7 +494,7 @@ class TestReadGraphML(BaseGraphML): nx.write_graphml(G, fh) fh.seek(0) H = nx.read_graphml(fh, node_type=int) - assert_edges_equal(G.edges(data=True, keys=True), H.edges(data=True, keys=True)) + assert edges_equal(G.edges(data=True, keys=True), H.edges(data=True, keys=True)) assert G._adj == H._adj Gadj = { @@ -1073,9 +1073,9 @@ class TestWriteGraphML(BaseGraphML): H = nx.read_graphml(fh) fh.seek(0) - assert_nodes_equal(G.nodes(), H.nodes()) - assert_edges_equal(G.edges(), H.edges()) - assert_edges_equal(G.edges(data=True), H.edges(data=True)) + assert nodes_equal(G.nodes(), H.nodes()) + assert edges_equal(G.edges(), H.edges()) + assert edges_equal(G.edges(data=True), H.edges(data=True)) self.attribute_named_key_ids_fh.seek(0) xml = parse(fh) @@ -1118,9 +1118,9 @@ class TestWriteGraphML(BaseGraphML): H = nx.read_graphml(fh) fh.seek(0) - assert_nodes_equal(G.nodes(), H.nodes()) - assert_edges_equal(G.edges(), H.edges()) - assert_edges_equal(G.edges(data=True), H.edges(data=True)) + assert nodes_equal(G.nodes(), H.nodes()) + assert edges_equal(G.edges(), H.edges()) + assert edges_equal(G.edges(data=True), H.edges(data=True)) self.attribute_numeric_type_fh.seek(0) xml = parse(fh) @@ -1146,7 +1146,7 @@ class TestWriteGraphML(BaseGraphML): self.writer(G, fname) H = nx.read_graphml(fname) assert H.is_multigraph() - assert_edges_equal(G.edges(keys=True), H.edges(keys=True)) + assert edges_equal(G.edges(keys=True), H.edges(keys=True)) assert G._adj == H._adj os.close(fd) os.unlink(fname) @@ -1162,8 +1162,8 @@ class TestWriteGraphML(BaseGraphML): self.writer(G, fh) fh.seek(0) H = nx.read_graphml(fh, node_type=int) - assert_nodes_equal(G.nodes(), H.nodes()) - assert_edges_equal(G.edges(), H.edges()) + assert nodes_equal(G.nodes(), H.nodes()) + assert edges_equal(G.edges(), H.edges()) assert G.graph == H.graph def test_mixed_type_attributes(self): |