fixed tests

Author: ashleymeigh2

mantidimaging/core/fitting/test/fitting_engine_test.py

@@ -41,6 +41,6 @@ def test_find_best_fit(self):
         xvals = np.linspace(1, 10, 20)
         yvals = 3.1 * xvals + 2.7
 
-        result = self.engine.find_best_fit(xvals, yvals, [1, 1])
-        self.assertAlmostEqual(result['a'], 3.1, 4)
-        self.assertAlmostEqual(result['b'], 2.7, 4)
+        fit_params, chi2 = self.engine.find_best_fit(xvals, yvals, [1, 1])
+        self.assertAlmostEqual(fit_params['a'], 3.1, 4)
+        self.assertAlmostEqual(fit_params['b'], 2.7, 4)

mantidimaging/gui/windows/spectrum_viewer/test/presenter_test.py

@@ -466,12 +466,15 @@ def test_on_initial_params_edited_updates_plot_correctly(self, is_initial_fit_vi
         self.view.spectrum_widget.get_roi = mock.Mock(return_value="mock_roi")
         self.presenter.model.get_spectrum = mock.Mock(return_value=np.array([1, 2, 3]))
         self.presenter.model.tof_data = np.array([1, 2, 3])
-        self.presenter.model.fitting_engine.find_best_fit = mock.Mock(return_value={
-            "mu": 1.0,
-            "sigma": 2.0,
-            "h": 3.0,
-            "a": 4.0
-        })
+        self.presenter.model.fitting_engine.find_best_fit = mock.Mock(return_value=(
+            {
+                "mu": 1.0,
+                "sigma": 2.0,
+                "h": 3.0,
+                "a": 4.0,
+            },
+            0.0,
+        ))
         self.view.scalable_roi_widget.set_fitted_parameter_values = mock.Mock()
         self.view.fittingDisplayWidget.is_initial_fit_visible.return_value = is_initial_fit_visible