File size: 4,973 Bytes
17c6d62 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 |
"""Unit tests for pydot drawing functions."""
from io import StringIO
import pytest
import networkx as nx
from networkx.utils import graphs_equal
pydot = pytest.importorskip("pydot")
class TestPydot:
@pytest.mark.parametrize("G", (nx.Graph(), nx.DiGraph()))
@pytest.mark.parametrize("prog", ("neato", "dot"))
def test_pydot(self, G, prog, tmp_path):
"""
Validate :mod:`pydot`-based usage of the passed NetworkX graph with the
passed basename of an external GraphViz command (e.g., `dot`, `neato`).
"""
# Set the name of this graph to... "G". Failing to do so will
# subsequently trip an assertion expecting this name.
G.graph["name"] = "G"
# Add arbitrary nodes and edges to the passed empty graph.
G.add_edges_from([("A", "B"), ("A", "C"), ("B", "C"), ("A", "D")])
G.add_node("E")
# Validate layout of this graph with the passed GraphViz command.
graph_layout = nx.nx_pydot.pydot_layout(G, prog=prog)
assert isinstance(graph_layout, dict)
# Convert this graph into a "pydot.Dot" instance.
P = nx.nx_pydot.to_pydot(G)
# Convert this "pydot.Dot" instance back into a graph of the same type.
G2 = G.__class__(nx.nx_pydot.from_pydot(P))
# Validate the original and resulting graphs to be the same.
assert graphs_equal(G, G2)
fname = tmp_path / "out.dot"
# Serialize this "pydot.Dot" instance to a temporary file in dot format
P.write_raw(fname)
# Deserialize a list of new "pydot.Dot" instances back from this file.
Pin_list = pydot.graph_from_dot_file(path=fname, encoding="utf-8")
# Validate this file to contain only one graph.
assert len(Pin_list) == 1
# The single "pydot.Dot" instance deserialized from this file.
Pin = Pin_list[0]
# Sorted list of all nodes in the original "pydot.Dot" instance.
n1 = sorted(p.get_name() for p in P.get_node_list())
# Sorted list of all nodes in the deserialized "pydot.Dot" instance.
n2 = sorted(p.get_name() for p in Pin.get_node_list())
# Validate these instances to contain the same nodes.
assert n1 == n2
# Sorted list of all edges in the original "pydot.Dot" instance.
e1 = sorted((e.get_source(), e.get_destination()) for e in P.get_edge_list())
# Sorted list of all edges in the original "pydot.Dot" instance.
e2 = sorted((e.get_source(), e.get_destination()) for e in Pin.get_edge_list())
# Validate these instances to contain the same edges.
assert e1 == e2
# Deserialize a new graph of the same type back from this file.
Hin = nx.nx_pydot.read_dot(fname)
Hin = G.__class__(Hin)
# Validate the original and resulting graphs to be the same.
assert graphs_equal(G, Hin)
def test_read_write(self):
G = nx.MultiGraph()
G.graph["name"] = "G"
G.add_edge("1", "2", key="0") # read assumes strings
fh = StringIO()
nx.nx_pydot.write_dot(G, fh)
fh.seek(0)
H = nx.nx_pydot.read_dot(fh)
assert graphs_equal(G, H)
def test_pydot_issue_7581(tmp_path):
"""Validate that `nx_pydot.pydot_layout` handles nodes
with characters like "\n", " ".
Those characters cause `pydot` to escape and quote them on output,
which caused #7581.
"""
G = nx.Graph()
G.add_edges_from([("A\nbig test", "B"), ("A\nbig test", "C"), ("B", "C")])
graph_layout = nx.nx_pydot.pydot_layout(G, prog="dot")
assert isinstance(graph_layout, dict)
# Convert the graph to pydot and back into a graph. There should be no difference.
P = nx.nx_pydot.to_pydot(G)
G2 = nx.Graph(nx.nx_pydot.from_pydot(P))
assert graphs_equal(G, G2)
@pytest.mark.parametrize(
"graph_type", [nx.Graph, nx.DiGraph, nx.MultiGraph, nx.MultiDiGraph]
)
def test_hashable_pydot(graph_type):
# gh-5790
G = graph_type()
G.add_edge("5", frozenset([1]), t='"Example:A"', l=False)
G.add_edge("1", 2, w=True, t=("node1",), l=frozenset(["node1"]))
G.add_edge("node", (3, 3), w="string")
assert [
{"t": '"Example:A"', "l": "False"},
{"w": "True", "t": "('node1',)", "l": "frozenset({'node1'})"},
{"w": "string"},
] == [
attr
for _, _, attr in nx.nx_pydot.from_pydot(nx.nx_pydot.to_pydot(G)).edges.data()
]
assert {str(i) for i in G.nodes()} == set(
nx.nx_pydot.from_pydot(nx.nx_pydot.to_pydot(G)).nodes
)
def test_pydot_numerical_name():
G = nx.Graph()
G.add_edges_from([("A", "B"), (0, 1)])
graph_layout = nx.nx_pydot.pydot_layout(G, prog="dot")
assert isinstance(graph_layout, dict)
assert "0" not in graph_layout
assert 0 in graph_layout
assert "1" not in graph_layout
assert 1 in graph_layout
assert "A" in graph_layout
assert "B" in graph_layout
|