1 | """ |
---|
2 | Unit tests for fitting module |
---|
3 | @author G.alina |
---|
4 | """ |
---|
5 | import unittest |
---|
6 | import math |
---|
7 | |
---|
8 | from sas.fit.AbstractFitEngine import Model |
---|
9 | from sas.fit.Fitting import Fit |
---|
10 | from sas.dataloader.loader import Loader |
---|
11 | |
---|
12 | class TestSingleFit(unittest.TestCase): |
---|
13 | """ test single fitting """ |
---|
14 | def setUp(self): |
---|
15 | """ initialize data""" |
---|
16 | self.data = Loader().load("cyl_400_20.txt") |
---|
17 | # Create model that fitting engine understands |
---|
18 | from sas.models.CylinderModel import CylinderModel |
---|
19 | self.model = CylinderModel() |
---|
20 | self.model.setParam("scale", 1.0) |
---|
21 | self.model.setParam("radius",18) |
---|
22 | self.model.setParam("length", 397) |
---|
23 | self.model.setParam("sldCyl",3e-006 ) |
---|
24 | self.model.setParam("sldSolv",0.0 ) |
---|
25 | self.model.setParam("background", 0.0) |
---|
26 | #select parameters to fit |
---|
27 | self.pars1 =['length','radius','scale'] |
---|
28 | |
---|
29 | def _fit(self, name="bumps"): |
---|
30 | """ return fit result """ |
---|
31 | fitter = Fit(name) |
---|
32 | fitter.set_data(self.data,1) |
---|
33 | fitter.set_model(self.model,1,self.pars1) |
---|
34 | fitter.select_problem_for_fit(id=1,value=1) |
---|
35 | result1, = fitter.fit() |
---|
36 | |
---|
37 | self.assert_(result1) |
---|
38 | self.assertTrue(len(result1.pvec)>0 or len(result1.pvec)==0 ) |
---|
39 | self.assertTrue(len(result1.stderr)> 0 or len(result1.stderr)==0) |
---|
40 | |
---|
41 | self.assertTrue( math.fabs(result1.pvec[0]-400.0)/3.0 < result1.stderr[0] ) |
---|
42 | self.assertTrue( math.fabs(result1.pvec[1]-20.0)/3.0 < result1.stderr[1] ) |
---|
43 | self.assertTrue( math.fabs(result1.pvec[2]-1.0)/3.0 < result1.stderr[2] ) |
---|
44 | self.assertTrue( result1.fitness < 1.0 ) |
---|
45 | |
---|
46 | |
---|
47 | def test_bumps(self): |
---|
48 | """ Simple cylinder model fit """ |
---|
49 | self._fit("bumps") |
---|
50 | |
---|
51 | |
---|
52 | |
---|
53 | class TestSimultaneousFit(unittest.TestCase): |
---|
54 | """ test simultaneous fitting """ |
---|
55 | def setUp(self): |
---|
56 | """ initialize data""" |
---|
57 | |
---|
58 | self.data1=Loader().load("cyl_400_20.txt") |
---|
59 | self.data2=Loader().load("cyl_400_40.txt") |
---|
60 | |
---|
61 | # Receives the type of model for the fitting |
---|
62 | from sas.models.CylinderModel import CylinderModel |
---|
63 | cyl1 = CylinderModel() |
---|
64 | cyl1.name = "C1" |
---|
65 | self.model1 = Model(cyl1) |
---|
66 | self.model1.set(scale= 1.0) |
---|
67 | self.model1.set(radius=18) |
---|
68 | self.model1.set(length=200) |
---|
69 | self.model1.set(sldCyl=3e-006, sldSolv=0.0) |
---|
70 | self.model1.set(background=0.0) |
---|
71 | |
---|
72 | cyl2 = CylinderModel() |
---|
73 | cyl2.name = "C2" |
---|
74 | self.model2 = Model(cyl2) |
---|
75 | self.model2.set(scale= 1.0) |
---|
76 | self.model2.set(radius=37) |
---|
77 | self.model2.set(length=300) |
---|
78 | self.model2.set(sldCyl=3e-006, sldSolv=0.0) |
---|
79 | self.model2.set(background=0.0) |
---|
80 | |
---|
81 | |
---|
82 | def test_constrained_bumps(self): |
---|
83 | """ Simultaneous cylinder model fit """ |
---|
84 | self._run_fit(Fit('bumps')) |
---|
85 | |
---|
86 | #@unittest.skip("") |
---|
87 | def _run_fit(self, fitter): |
---|
88 | result1, result2 = self._fit(fitter) |
---|
89 | self.assert_(result1) |
---|
90 | self.assertTrue(len(result1.pvec)>0) |
---|
91 | self.assertTrue(len(result1.stderr)>0) |
---|
92 | |
---|
93 | for n, v, dv in zip(result1.param_list, result1.pvec, result1.stderr): |
---|
94 | #print "%s M1.%s = %s +/- %s"%(fitter._engine.__class__.__name__,n,v,dv) |
---|
95 | if n == "length": |
---|
96 | self.assertTrue( math.fabs(v-400.0)/3.0 < dv ) |
---|
97 | elif n=='radius': |
---|
98 | self.assertTrue( math.fabs(v-20.0)/3.0 < dv ) |
---|
99 | elif n=='scale': |
---|
100 | self.assertTrue( math.fabs(v-1.0)/3.0 < dv ) |
---|
101 | for n, v, dv in zip(result2.param_list, result2.pvec, result2.stderr): |
---|
102 | #print "%s M2.%s = %s +/- %s"%(fitter._engine.__class__.__name__,n,v,dv) |
---|
103 | if n=='radius': |
---|
104 | self.assertTrue( math.fabs(v-40.0)/3.0 < dv ) |
---|
105 | elif n=='scale': |
---|
106 | self.assertTrue( math.fabs(v-1.0)/3.0 < dv ) |
---|
107 | |
---|
108 | def _fit(self, fitter): |
---|
109 | """ return fit result """ |
---|
110 | fitter.set_data(self.data1,1) |
---|
111 | fitter.set_model(self.model1, 1, ['length','radius','scale']) |
---|
112 | |
---|
113 | fitter.set_data(self.data2,2) |
---|
114 | fitter.set_model(self.model2, 2, ['length','radius','scale'], |
---|
115 | constraints=[("length","C1.length")]) |
---|
116 | fitter.select_problem_for_fit(id=1,value=1) |
---|
117 | fitter.select_problem_for_fit(id=2,value=1) |
---|
118 | return fitter.fit() |
---|
119 | |
---|
120 | |
---|
121 | if __name__ == '__main__': |
---|
122 | unittest.main() |
---|
123 | |
---|