* Patched "plt.show()" calls. #41

garciparedes · Sep 1, 2019 · fd693d6 · fd693d6
1 parent 3c23f1a
commit fd693d6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/tests/test_storers/test_graph.py b/tests/test_storers/test_graph.py
@@ -1,5 +1,6 @@
 import unittest
 from pathlib import Path
+from unittest.mock import patch
 
 import networkx as nx
 
@@ -25,7 +26,8 @@ def test_creation(self):
         )
         self.assertEqual(storer.result, self.result)
 
-    def test_store(self):
+    @patch("jinete.storers.plots.graph.plt.show")
+    def test_store(self, mocked_plt):
         storer = jit.GraphPlotStorer(
             result=self.result,
         )

diff --git a/tests/test_storers/test_set.py b/tests/test_storers/test_set.py
@@ -1,5 +1,6 @@
 import unittest
 from pathlib import Path
+from unittest.mock import patch
 
 import jinete as jit
 
@@ -29,7 +30,8 @@ def test_creation(self):
         self.assertEqual(storer.result, self.result)
         self.assertEqual(storer.storer_cls_set, storer_cls_set)
 
-    def test_store(self):
+    @patch("jinete.storers.plots.graph.plt.show")
+    def test_store(self, mocked_plt):
         storer = jit.StorerSet(
             result=self.result,
             storer_cls_set={