Fix a high priority bug where reconstructions will crash upon saving, and add test coverage to prevent a repeat

tests/models/test_fancy_ptycho.py

@@ -123,6 +123,9 @@ def test_lab_ptycho(lab_ptycho_cxi, reconstruction_device, show_plot):
         time.sleep(3)
         plt.close('all')
 
+    # Simply test that this does not fail
+    results = model.save_results(dataset)
+
     # If this fails, the reconstruction has gotten worse
     assert model.loss_history[-1] < 0.38
 

tests/models/test_simple_ptycho.py

@@ -29,5 +29,8 @@ def test_simple_ptycho(lab_ptycho_cxi, reconstruction_device, show_plot):
         time.sleep(3)
         plt.close('all')
 
+    # Simply test that this does not fail
+    results = model.save_results(dataset)
+
     # If this fails, the reconstruction got worse
     assert model.loss_history[-1] < 6.5