def test_filename(self):
graph = retworkx.generators.grid_graph(20, 20)
graphviz_draw(
graph,
filename="test_graphviz_filename.svg",
image_type="svg",
method="neato",
)
self.assertTrue(os.path.isfile("test_graphviz_filename.svg"))
if not SAVE_IMAGES:
self.addCleanup(os.remove, "test_graphviz_filename.svg")
def test_draw_graph_attr(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"))
graph_attr = {"bgcolor": "red"}
image = graphviz_draw(
graph, lambda node: node, lambda edge: edge, graph_attr
)
self.assertIsInstance(image, PIL.Image.Image)
_save_image(image, "test_graphviz_draw_graph_attr.png")
def test_draw_no_args(self):
graph = retworkx.generators.star_graph(24)
image = graphviz_draw(graph)
self.assertIsInstance(image, PIL.Image.Image)
_save_image(image, "test_graphviz_draw.png")
def test_method(self):
graph = retworkx.directed_gnp_random_graph(50, 0.8)
image = graphviz_draw(graph, method="sfdp")
self.assertIsInstance(image, PIL.Image.Image)
_save_image(image, "test_graphviz_method.png")
def test_image_type(self):
graph = retworkx.directed_gnp_random_graph(50, 0.8)
image = graphviz_draw(graph, image_type="jpg")
self.assertIsInstance(image, PIL.Image.Image)
_save_image(image, "test_graphviz_draw_image_type.jpg")