summaryrefslogtreecommitdiff
path: root/networkx/readwrite/tests/test_graphml.py
diff options
context:
space:
mode:
authorJarrod Millman <jarrod.millman@gmail.com>2021-05-26 08:59:54 -0700
committerGitHub <noreply@github.com>2021-05-26 08:59:54 -0700
commit09b30ba69dd2e81a2728fd1259ca7b4d88470827 (patch)
treeec1c723f51473a8ff99fb0195783c607242bb90c /networkx/readwrite/tests/test_graphml.py
parent7771b28bad07f0a07ac08ef8c2c6a17191d36f54 (diff)
downloadnetworkx-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.py56
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):