Slight adjustment to make testing easier

Author: pshriwise

openmc_plotter/main_window.py

@@ -479,14 +479,15 @@ def loadModel(self, reload=False, use_settings_pkl=True):
             self.plotIm.model = self.model
             self.applyChanges()
 
-    def saveImage(self):
-        filename, ext = QFileDialog.getSaveFileName(self,
-                                                    "Save Plot Image",
-                                                    "untitled",
-                                                    "Images (*.png)")
+    def saveImage(self, filename=None):
+        if filename is None:
+            filename, ext = QFileDialog.getSaveFileName(self,
+                                                        "Save Plot Image",
+                                                        "untitled",
+                                                        "Images (*.png)")
         if filename:
             self.plotIm.saveImage(filename)
-        self.statusBar().showMessage('Plot Image Saved', 5000)
+            self.statusBar().showMessage('Plot Image Saved', 5000)
 
     def saveView(self):
         filename, ext = QFileDialog.getSaveFileName(self,

openmc_plotter/plotgui.py

@@ -119,7 +119,7 @@ def saveImage(self, filename):
         filename : str or pathlib.Path
             Name of the image to save
         """
-        if "." not in filename:
+        if "." not in str(filename):
             filename += ".png"
         self.figure.savefig(filename, transparent=True)
 

tests/setup_test/test.py

@@ -1,9 +1,23 @@
+import filecmp
+import pytest
 from openmc_plotter.main_window import MainWindow, _openmcReload
 
-def test_window(qtbot):
-    _openmcReload()
+@pytest.fixture
+def run_in_tmpdir(tmpdir):
+    orig = tmpdir.chdir()
+    try:
+        yield
+    finally:
+        orig.chdir()
 
-    mw = MainWindow()
+def test_window(tmpdir, qtbot):
+    orig = tmpdir.chdir()
+
+    _openmcReload(model_path=str(orig))
+
+    mw = MainWindow(model_path=orig)
     mw.loadGui()
-    mw.plotIm.figure.savefig("test.png")
+    mw.saveImage(tmpdir / 'test.png')
     qtbot.addWidget(mw)
+
+    filecmp.cmp(orig / 'ref.png', tmpdir / 'test.png')