Adding another test reference and making some small tweaks to polish the feature.

Author: pshriwise

openmc_plotter/__main__.py

@@ -46,7 +46,8 @@ def run_app(user_args):
     splash_pix = QtGui.QPixmap(path_splash)
     splash = QSplashScreen(splash_pix, QtCore.Qt.WindowStaysOnTopHint)
     splash.setMask(splash_pix.mask())
-    splash.show()
+    if not user_args.batch_mode:
+        splash.show()
     app.processEvents()
     splash.setMask(splash_pix.mask())
     splash.showMessage("Loading Model...",

openmc_plotter/main_window.py

@@ -159,7 +159,8 @@ def createMenuBar(self):
         self.saveImageAction.setShortcut("Ctrl+Shift+S")
         self.saveImageAction.setToolTip('Save plot image')
         self.saveImageAction.setStatusTip('Save plot image')
-        self.saveImageAction.triggered.connect(self.saveImage)
+        save_image_connector = partial(self.saveImage, filename=None)
+        self.saveImageAction.triggered.connect(save_image_connector)
 
         self.saveViewAction = QAction("Save &View...", self)
         self.saveViewAction.setShortcut(QtGui.QKeySequence.Save)

tests/setup_test/ref1.png

@@ -21,27 +21,37 @@ def test_window(tmpdir, qtbot):
 
     try:
         mw.saveImage(tmpdir / 'test.png')
-        mw.close()
+
         qtbot.addWidget(mw)
     finally:
         orig.chdir()
 
     filecmp.cmp(orig / 'ref.png', tmpdir / 'test.png')
 
+    mw.close()
+
 def test_batch_image(tmpdir, qtbot):
     orig = tmpdir.chdir()
 
     # move view file into tmpdir
     shutil.copy2(orig / 'test.pltvw', tmpdir)
+    shutil.copy2(orig / 'test1.pltvw', tmpdir)
 
     _openmcReload(model_path=orig)
 
     mw = MainWindow(model_path=orig)
     mw.loadGui()
+
     try:
         mw.saveBatchImage('test.pltvw')
         qtbot.addWidget(mw)
+
+        mw.saveBatchImage('test1.pltvw')
+        qtbot.addWidget(mw)
     finally:
         orig.chdir()
 
     filecmp.cmp(orig / 'ref.png', tmpdir / 'test.png')
+    filecmp.cmp(orig / 'ref1.png', tmpdir / 'test1.png')
+
+    mw.close()