| 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 | |
|---|
