retworkx 0.8.0

A python graph library implemented in Rust
Documentation 
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

# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.

import os
import tempfile
import unittest

import retworkx


class TestDot(unittest.TestCase):
    def setUp(self):
        fd, self.path = tempfile.mkstemp()
        os.close(fd)
        os.remove(self.path)

    def test_graph_to_dot(self):
        graph = retworkx.PyGraph()
        graph.add_node({'color': 'black', 'fillcolor': 'green',
                        'label': "a", 'style': 'filled'})
        graph.add_node({'color': 'black', 'fillcolor': 'red',
                        'label': "a", 'style': 'filled'})
        graph.add_edge(0, 1, dict(label='1', name='1'))
        expected = (
            'graph {\n0 [color=black, fillcolor=green, label="a", style=filled'
            '];\n1 [color=black, fillcolor=red, label="a", style=filled];'
            '\n0 -- 1 [label="1", name=1];\n}\n')
        res = graph.to_dot(lambda node: node, lambda edge: edge)
        self.assertEqual(expected, res)

    def test_digraph_to_dot(self):
        graph = retworkx.PyDiGraph()
        graph.add_node({'color': 'black', 'fillcolor': 'green',
                        'label': "a", 'style': 'filled'})
        graph.add_node({'color': 'black', 'fillcolor': 'red',
                        'label': "a", 'style': 'filled'})
        graph.add_edge(0, 1, dict(label='1', name='1'))
        expected = (
            'digraph {\n0 [color=black, fillcolor=green, label="a", '
            'style=filled];\n1 [color=black, fillcolor=red, label="a", '
            'style=filled];\n0 -> 1 [label="1", name=1];\n}\n')
        res = graph.to_dot(lambda node: node, lambda edge: edge)
        self.assertEqual(expected, res)

    def test_graph_to_dot_to_file(self):
        graph = retworkx.PyGraph()
        graph.add_node({'color': 'black', 'fillcolor': 'green',
                        'label': "a", 'style': 'filled'})
        graph.add_node({'color': 'black', 'fillcolor': 'red',
                        'label': "a", 'style': 'filled'})
        graph.add_edge(0, 1, dict(label='1', name='1'))
        expected = (
            'graph {\n0 [color=black, fillcolor=green, label="a", '
            'style=filled];\n1 [color=black, fillcolor=red, label="a", '
            'style=filled];\n0 -- 1 [label="1", name=1];\n}\n')
        res = graph.to_dot(lambda node: node, lambda edge: edge,
                           filename=self.path)
        self.addCleanup(os.remove, self.path)
        self.assertIsNone(res)
        with open(self.path, 'r') as fd:
            res = fd.read()
        self.assertEqual(expected, res)

    def test_digraph_to_dot_to_file(self):
        graph = retworkx.PyDiGraph()
        graph.add_node({'color': 'black', 'fillcolor': 'green',
                        'label': "a", 'style': 'filled'})
        graph.add_node({'color': 'black', 'fillcolor': 'red',
                        'label': "a", 'style': 'filled'})
        graph.add_edge(0, 1, dict(label='1', name='1'))
        expected = (
            'digraph {\n0 [color=black, fillcolor=green, label="a", '
            'style=filled];\n1 [color=black, fillcolor=red, label="a", '
            'style=filled];\n0 -> 1 [label="1", name=1];\n}\n')
        res = graph.to_dot(lambda node: node, lambda edge: edge,
                           filename=self.path)
        self.addCleanup(os.remove, self.path)
        self.assertIsNone(res)
        with open(self.path, 'r') as fd:
            res = fd.read()
        self.assertEqual(expected, res)

    def test_digraph_empty_dicts(self):
        graph = retworkx.directed_gnp_random_graph(3, .9, seed=42)
        dot_str = graph.to_dot(lambda _: {}, lambda _: {})
        self.assertEqual("digraph {\n0 ;\n1 ;\n2 ;\n0 -> 1 ;\n0 -> 2 ;\n}\n",
                         dot_str)

    def test_graph_empty_dicts(self):
        graph = retworkx.undirected_gnp_random_graph(3, .9, seed=42)
        dot_str = graph.to_dot(lambda _: {}, lambda _: {})
        self.assertEqual("graph {\n0 ;\n1 ;\n2 ;\n1 -- 0 ;\n2 -- 0 ;\n"
                         "2 -- 1 ;\n}\n", dot_str)

    def test_digraph_graph_attrs(self):
        graph = retworkx.directed_gnp_random_graph(3, .9, seed=42)
        dot_str = graph.to_dot(lambda _: {}, lambda _: {}, {'bgcolor': 'red'})
        self.assertEqual("digraph {\nbgcolor=red ;\n0 ;\n1 ;\n2 ;\n0 -> 1 ;\n"
                         "0 -> 2 ;\n}\n", dot_str)

    def test_graph_graph_attrs(self):
        graph = retworkx.undirected_gnp_random_graph(3, .9, seed=42)
        dot_str = graph.to_dot(lambda _: {}, lambda _: {}, {'bgcolor': 'red'})
        self.assertEqual("graph {\nbgcolor=red ;\n0 ;\n1 ;\n2 ;\n1 -- 0 ;\n"
                         "2 -- 0 ;\n2 -- 1 ;\n}\n", dot_str)

    def test_graph_no_args(self):
        graph = retworkx.undirected_gnp_random_graph(3, .95, seed=24)
        dot_str = graph.to_dot()
        self.assertEqual("graph {\n0 ;\n1 ;\n2 ;\n2 -- 0 ;\n2 -- 1 ;\n}\n",
                         dot_str)

    def test_digraph_no_args(self):
        graph = retworkx.directed_gnp_random_graph(3, .95, seed=24)
        dot_str = graph.to_dot()
        self.assertEqual("digraph {\n0 ;\n1 ;\n2 ;\n0 -> 1 ;\n0 -> 2 ;\n}\n",
                         dot_str)