source: sasview/park_integration/test/testnewscipys.py @ 48882d1

"""
    Unit tests for fitting module 
"""
import unittest
from sans.guitools.plottables import Theory1D
from sans.guitools.plottables import Data1D
from sans.fit.AbstractFitEngine import Data, Model
from sans.fit.AbstractFitEngine import sansAssembly
import math
class testFitModule(unittest.TestCase):
    """ test fitting """
    
    def test_Data(self):
        """ test data"""
        #load data
        from sans.fit.Loader import Load
        load= Load()
        load.set_filename("cyl_testdata.txt")
        load.set_values()
        data11 = Data1D(x=[], y=[],dx=None, dy=None)
        load.load_data(data11)
        
        #Importing the Fit module
        from sans.fit.Fitting import Fit
        fitter= Fit('scipy')
        
        # Receives the type of model for the fitting
        from sans.guitools.LineModel import LineModel
        model1  = LineModel()
        data1=Data(sans_data=data11)
        model =Model(model1)
        #Do the fit SCIPY0.007778 1.78E+14 13335629 
        v=[0.007778,0.023333,0.031111,0.038889,0.046667,0.054444]
        for i in range(len(v)):
            self.assertEquals(data1.x[i],v[i])
        #fitter.set_data_assembly(data1,1)
        #fitter.set_model_assembly(model,"M1",1,['A','B'])
    def test_Model(self):
        """ test model"""
        #load data
        from sans.fit.Loader import Load
        load= Load()
        load.set_filename("cyl_testdata.txt")
        load.set_values()
        data11 = Data1D(x=[], y=[],dx=None, dy=None)
        load.load_data(data11)
        
        #Importing the Fit module
        from sans.fit.Fitting import Fit
        fitter= Fit('scipy')
        
        # Receives the type of model for the fitting
        from sans.models.CylinderModel import CylinderModel
        model1  = CylinderModel()
        data1=Data(sans_data=data11)
        model =Model(model1)
        for item in data1.x:
            self.assertEquals(model.eval(item),model.model.run(item))
        p=[0.0, 3.0000000000000001e-006, 1.0, 1.0, 400.0, 20.0, 1.0]  
        model.setParams(p)
        res=[]
        for i in range(len(data1.x)):
            res.append((data1.y[i]-model.eval(data1.x[i]))/data1.dy[i])
            self.assertEquals(model.eval(data1.x[i]),model.model.run(data1.x[i]))  
        print res
    def test_funtor(self):
        """ test functor"""
        #load data
        from sans.fit.Loader import Load
        load= Load()
        load.set_filename("cyl_testdata.txt")
        load.set_values()
        data11 = Data1D(x=[], y=[],dx=None, dy=None)
        load.load_data(data11)
        
        #Importing the Fit module
        from sans.fit.Fitting import Fit
        fitter= Fit('scipy')
        
        # Receives the type of model for the fitting
        from sans.models.CylinderModel import CylinderModel
        model1  = CylinderModel()
        data1=Data(sans_data=data11)
        model =Model(model1)
        functor= sansAssembly(model,data1)
        for item in data1.x:
            self.assertEquals(model.eval(item),model.model.run(item))
        p=[0.0, 3.0000000000000001e-006, 1.0, 1.0, 400.0, 20.0, 1.0]  
        model.setParams(p)
        res=[]
        for i in range(len(data1.x)):
            res.append((data1.y[i]-model.eval(data1.x[i]))/data1.dy[i])
            self.assertEquals(model.eval(data1.x[i]),model.model.run(data1.x[i]))  
        f=functor(p)

        for i in range(len(res)):
            self.assertEquals(res[i],f[i])