Adding test for batch image function

Author: pshriwise

openmc_plotter/__main__.py

@@ -75,7 +75,6 @@ def run_app(user_args):
 
     if user_args.batch_mode:
         for view_file in user_args.batch_mode:
-            print(view_file)
             mainWindow.saveBatchImage(view_file)
         mainWindow.close()
         sys.exit()

openmc_plotter/main_window.py

@@ -36,7 +36,7 @@ def _openmcReload(threads=None, model_path='.'):
     args = ["-c"]
     if threads is not None:
         args += ["-s", str(threads)]
-    args.append(model_path)
+    args.append(str(model_path))
     openmc.lib.init(args)
     openmc.lib.settings.verbosity = 1
 
@@ -450,6 +450,14 @@ def updateWindowMenu(self):
         self.mainWindowAction.setChecked(self.isActiveWindow())
 
     def saveBatchImage(self, view_file):
+        """
+        Loads a view in the GUI and generates an image
+
+        Parameters
+        ----------
+        view_file : str or pathlib.Path
+            The path to a view file that is compatible with the loaded model.
+        """
         # store the
         cv = self.model.currentView
         # load the view from file

tests/setup_test/test.pltvw

@@ -1,5 +1,8 @@
 import filecmp
+import shutil
+
 import pytest
+
 from openmc_plotter.main_window import MainWindow, _openmcReload
 
 @pytest.fixture
@@ -12,12 +15,33 @@ def run_in_tmpdir(tmpdir):
 
 def test_window(tmpdir, qtbot):
     orig = tmpdir.chdir()
+    mw = MainWindow(model_path=orig)
+    _openmcReload(model_path=orig)
+    mw.loadGui()
+
+    try:
+        mw.saveImage(tmpdir / 'test.png')
+        mw.close()
+        qtbot.addWidget(mw)
+    finally:
+        orig.chdir()
 
-    _openmcReload(model_path=str(orig))
+    filecmp.cmp(orig / 'ref.png', tmpdir / 'test.png')
+
+def test_batch_image(tmpdir, qtbot):
+    orig = tmpdir.chdir()
+
+    # move view file into tmpdir
+    shutil.copy2(orig / 'test.pltvw', tmpdir)
+
+    _openmcReload(model_path=orig)
 
     mw = MainWindow(model_path=orig)
     mw.loadGui()
-    mw.saveImage(tmpdir / 'test.png')
-    qtbot.addWidget(mw)
+    try:
+        mw.saveBatchImage('test.pltvw')
+        qtbot.addWidget(mw)
+    finally:
+        orig.chdir()
 
-    filecmp.cmp(orig / 'ref.png', tmpdir / 'test.png')
+    filecmp.cmp(orig / 'ref.png', tmpdir / 'test.png')