From b19d02240595f819817b1157945911d701a8e3a0 Mon Sep 17 00:00:00 2001 From: Samuel Genheden Date: Mon, 14 Jun 2021 06:37:43 +0200 Subject: [PATCH] Fix failing tests --- tests/test_image.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/test_image.py b/tests/test_image.py index 04da9ef..9efd330 100644 --- a/tests/test_image.py +++ b/tests/test_image.py @@ -66,6 +66,8 @@ def test_graphviz_usage(mocker, tmpdir, setup_graphviz_graph): files = [ (None, str(tmpdir / "graph1.dot")), (None, str(tmpdir / "img2.png")), + (None, str(tmpdir / "graph1.dot")), + (None, str(tmpdir / "img2.png")), ] mkstemp_patch.side_effect = files molecules, reactions, edges, frame_colors = setup_graphviz_graph @@ -80,7 +82,7 @@ def test_graphviz_usage(mocker, tmpdir, setup_graphviz_graph): def test_graphviz_usage_exception_dot(mocker, tmpdir, setup_graphviz_graph): exists_patch = mocker.patch("aizynthfinder.utils.image.os.path.exists") - exists_patch.side_effect = [False, True] + exists_patch.side_effect = [False, True, True, True] molecules, reactions, edges, frame_colors = setup_graphviz_graph img = image.make_graphviz_image(molecules, reactions, edges, frame_colors)