| 1 | """ |
|---|
| 2 | Unit tests for fitting module |
|---|
| 3 | """ |
|---|
| 4 | import unittest |
|---|
| 5 | from sans.guitools.plottables import Theory1D |
|---|
| 6 | from sans.guitools.plottables import Data1D |
|---|
| 7 | from FittingModule import Parameter |
|---|
| 8 | import math |
|---|
| 9 | class testFitModule(unittest.TestCase): |
|---|
| 10 | """ test fitting """ |
|---|
| 11 | def testLoader(self): |
|---|
| 12 | """ |
|---|
| 13 | test module Load |
|---|
| 14 | """ |
|---|
| 15 | from Loader import Load |
|---|
| 16 | load= Load() |
|---|
| 17 | |
|---|
| 18 | load.set_filename("testdata_line.txt") |
|---|
| 19 | self.assertEqual(load.get_filename(),"testdata_line.txt") |
|---|
| 20 | load.set_values() |
|---|
| 21 | x=[] |
|---|
| 22 | y=[] |
|---|
| 23 | dx=[] |
|---|
| 24 | dy=[] |
|---|
| 25 | |
|---|
| 26 | x,y,dx,dy = load.get_values() |
|---|
| 27 | |
|---|
| 28 | # test that values have been loaded |
|---|
| 29 | self.assertNotEqual(x, None) |
|---|
| 30 | self.assertNotEqual(y, []) |
|---|
| 31 | self.assertNotEqual(dy, None) |
|---|
| 32 | self.assertEqual(len(x),len(y)) |
|---|
| 33 | self.assertEqual(len(dy),len(y)) |
|---|
| 34 | |
|---|
| 35 | # test data the two plottables contained values loaded |
|---|
| 36 | data1 = Theory1D(x=[], y=[], dy=None) |
|---|
| 37 | data2 = Data1D(x=[], y=[],dx=None, dy=None) |
|---|
| 38 | data1.name = "data1" |
|---|
| 39 | data2.name = "data2" |
|---|
| 40 | load.load_data(data1) |
|---|
| 41 | load.load_data(data2) |
|---|
| 42 | |
|---|
| 43 | |
|---|
| 44 | for i in range(len(x)): |
|---|
| 45 | self.assertEqual(data2.x[i],x[i]) |
|---|
| 46 | self.assertEqual(data1.y[i],y[i]) |
|---|
| 47 | self.assertEqual(data2.y[i],y[i]) |
|---|
| 48 | self.assertEqual(data1.dx[i],dx[i]) |
|---|
| 49 | self.assertEqual(data2.dy[i],dy[i]) |
|---|
| 50 | self.assertEqual(data1.x[i],data2.x[i]) |
|---|
| 51 | self.assertEqual(data2.y[i],data2.y[i]) |
|---|
| 52 | |
|---|
| 53 | |
|---|
| 54 | def testfit(self): |
|---|
| 55 | """ test fitting""" |
|---|
| 56 | from Loader import Load |
|---|
| 57 | load= Load() |
|---|
| 58 | load.set_filename("testdata_line.txt") |
|---|
| 59 | load.set_values() |
|---|
| 60 | x,y,dx,dy = load.get_values() |
|---|
| 61 | data1 = Data1D(x=[], y=[],dx=None, dy=None) |
|---|
| 62 | load.load_data(data1) |
|---|
| 63 | |
|---|
| 64 | # Receives the type of model for the fitting |
|---|
| 65 | from sans.guitools.LineModel import LineModel |
|---|
| 66 | model = LineModel() |
|---|
| 67 | |
|---|
| 68 | from FittingModule import Fitting |
|---|
| 69 | Fit= Fitting() |
|---|
| 70 | Fit.set_data(data1) |
|---|
| 71 | Fit.set_model(model) |
|---|
| 72 | |
|---|
| 73 | default_A = model.getParam('A') |
|---|
| 74 | default_B = model.getParam('B') |
|---|
| 75 | cstA = Parameter(model, 'A', default_A) |
|---|
| 76 | cstB = Parameter(model, 'B', default_B) |
|---|
| 77 | chisqr, out, cov=Fit.fit([cstA,cstB],None,None) |
|---|
| 78 | print"fit only one data",chisqr, out, cov |
|---|
| 79 | |
|---|
| 80 | self.assertEqual(Fit.fit_engine(),True) |
|---|
| 81 | |
|---|
| 82 | self.assert_(math.fabs(out[1]-2.5)/math.sqrt(cov[1][1]) < 2) |
|---|
| 83 | print "chisqr",chisqr/len(data1.x) |
|---|
| 84 | self.assert_(chisqr/len(data1.x) < 2) |
|---|
| 85 | # self.assertAlmostEqual(out[1],2.5) |
|---|
| 86 | #self.assertAlmostEquals(out[0],4.0) |
|---|
| 87 | #self.assertAlmostEquals(out[1]+cov[1][1],2.5) |
|---|
| 88 | #self.assertAlmostEquals(out[0]+cov[0][0],4.0) |
|---|
| 89 | |
|---|
