Skip to content

Commit

Permalink
Cleaned last test for fit using Minos
Browse files Browse the repository at this point in the history
  • Loading branch information
GuiMacielPereira committed Jan 9, 2024
1 parent b9c1156 commit bb7118b
Showing 1 changed file with 22 additions and 44 deletions.
66 changes: 22 additions & 44 deletions mvesuvio/system_tests/test_yspace_fit_GC.py
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,10 @@ def _run(cls):
bootIC = BootstrapInitialConditions
userCtr = UserScriptControls
yFitIC.fitModel = "GC_C4_C6"
yFitIC.symmetrisationFlag = False

cls.load_workspaces()

scattRes, yfitRes = runScript(
userCtr, scriptName, wsBackIC, wsFrontIC, bckwdIC, fwdIC, yFitIC, bootIC, True
)
Expand All @@ -91,92 +94,67 @@ def setUpClass(cls):

cls.optdataY = AnalysisRunner.get_current_result().YSpaceSymSumDataY
cls.optdataE = AnalysisRunner.get_current_result().YSpaceSymSumDataE
cls.rtol = 0.000001
cls.equal_nan = True
cls.decimal = 6

def xtest_YSpaceDataY(self):
nptest.assert_allclose(self.oridataY, self.optdataY)
def test_YSpaceDataY(self):
nptest.assert_almost_equal(self.oridataY, self.optdataY)

def xtest_YSpaceDataE(self):
nptest.assert_allclose(self.oridataE, self.optdataE)
def test_YSpaceDataE(self):
nptest.assert_almost_equal(self.oridataE, self.optdataE)


class TestResolution(unittest.TestCase):
def setUp(self):
self.orires = AnalysisRunner.get_benchmark_result()["resolution"]
self.optres = AnalysisRunner.get_current_result().resolution

self.rtol = 0.0001
self.equal_nan = True
self.decimal = 8

def xtest_resolution(self):
nptest.assert_array_equal(self.orires, self.optres)
def test_resolution(self):
nptest.assert_almost_equal(self.orires, self.optres)


class TestHdataY(unittest.TestCase):
def setUp(self):
self.oriHdataY = AnalysisRunner.get_benchmark_result()["HdataY"]
self.optHdataY = AnalysisRunner.get_current_result().HdataY

self.rtol = 0.0001
self.equal_nan = True
self.decimal = 4

def xtest_HdataY(self):
# mask = np.isclose(self.oriHdataY, self.optHdataY, rtol=1e-9)
# plt.imshow(mask, aspect="auto", cmap=plt.cm.RdYlGn,
# interpolation="nearest", norm=None)
# plt.show()
nptest.assert_array_equal(self.oriHdataY, self.optHdataY)
def test_HdataY(self):
nptest.assert_almost_equal(self.oriHdataY, self.optHdataY)


class TestFinalRawDataY(unittest.TestCase):
def setUp(self):
self.oriFinalDataY = AnalysisRunner.get_benchmark_result()["finalRawDataY"]
self.optFinalDataY = AnalysisRunner.get_current_result().finalRawDataY

self.rtol = 1e-6
self.equal_nan = True
self.decimal = 10

def xtest_FinalDataY(self):
nptest.assert_array_equal(self.oriFinalDataY, self.optFinalDataY)
def test_FinalDataY(self):
nptest.assert_almost_equal(self.oriFinalDataY, self.optFinalDataY)


class TestFinalRawDataE(unittest.TestCase):
def setUp(self):
self.oriFinalDataE = AnalysisRunner.get_benchmark_result()["finalRawDataE"]
self.optFinalDataE = AnalysisRunner.get_current_result().finalRawDataE

self.rtol = 1e-6
self.equal_nan = True
self.decimal = 10

def xtest_HdataE(self):
nptest.assert_array_equal(self.oriFinalDataE, self.optFinalDataE)
def test_HdataE(self):
nptest.assert_almost_equal(self.oriFinalDataE, self.optFinalDataE)


class Testpopt(unittest.TestCase):
def setUp(self):
self.oripopt = AnalysisRunner.get_benchmark_result()["popt"]
# Select only Fit results due to Mantid Fit
self.optpopt = AnalysisRunner.get_current_result().popt
self.oripopt = AnalysisRunner.get_benchmark_result()["popt"][:, :-1] # Exclude Chi2 from test
self.optpopt = AnalysisRunner.get_current_result().popt[:, :-1]

def xtest_opt(self):
print("\nori:\n", self.oripopt, "\nopt:\n", self.optpopt)
nptest.assert_array_equal(self.oripopt, self.optpopt)
def test_opt(self):
print(self.oripopt, self.optpopt)
nptest.assert_almost_equal(self.oripopt, self.optpopt)


class Testperr(unittest.TestCase):
def setUp(self):
self.oriperr = AnalysisRunner.get_benchmark_result()["perr"]
self.optperr = AnalysisRunner.get_current_result().perr

def xtest_perr(self):
# print("\norierr:\n", self.oriperr, "\nopterr:\n", self.optperr)
nptest.assert_array_equal(self.oriperr, self.optperr)
def test_perr(self):
nptest.assert_almost_equal(self.oriperr, self.optperr, decimal=5)


if __name__ == "__main__":
Expand Down

0 comments on commit bb7118b

Please sign in to comment.