Changeset 97603c0 in sasview for Invariant/test

Timestamp:

Jan 20, 2010 5:05:16 PM (15 years ago)

Author:

Gervaise Alina <gervyh@…>

Branches:

master, ESS_GUI, ESS_GUI_Docs, ESS_GUI_batch_fitting, ESS_GUI_bumps_abstraction, ESS_GUI_iss1116, ESS_GUI_iss879, ESS_GUI_iss959, ESS_GUI_opencl, ESS_GUI_ordering, ESS_GUI_sync_sascalc, costrafo411, magnetic_scatt, release-4.1.1, release-4.1.2, release-4.2.2, release_4.0.1, ticket-1009, ticket-1094-headless, ticket-1242-2d-resolution, ticket-1243, ticket-1249, ticket885, unittest-saveload

Children:

889d5ab4

Parents:

662da312

Message:

add test to evaluate the output of the fit for slit smear data

File:

• Invariant/test/utest_data_handling.py (modified) (2 diffs)

Invariant/test/utest_data_handling.py

@@ -13 +13 @@
 from DataLoader.data_info import Data1D
 from sans.invariant import invariant
+from DataLoader.qsmearing import smear_selection
     
 class TestLinearFit(unittest.TestCase):
@@ -206 +207 @@
         self.assertEqual(len(y_out), 3)
         self.assertEqual(len(dy_out), 3)
-    
+
+    
+class TestDataExtraLow(unittest.TestCase):
+    """
+        Generate a Guinier distribution and verify that the extrapolation
+        produce the correct ditribution. Tested if the data generated by the 
+        invariant calculator is correct
+    """
+    
+    def setUp(self):
+        """
+            Generate a Guinier distribution. After extrapolating, we will
+            verify that we obtain the scale and rg parameters
+        """
+        self.scale = 1.5
+        self.rg = 30.0
+        x = numpy.arange(0.0001, 0.1, 0.0001)
+        y = numpy.asarray([self.scale * math.exp( -(q*self.rg)**2 / 3.0 ) for q in x])
+        dy = y*.1
+        self.data = Data1D(x=x, y=y, dy=dy)
+        
+    def test_low_q(self):
+        """
+            Invariant with low-Q extrapolation with no slit smear
+        """
+        # Create invariant object. Background and scale left as defaults.
+        inv = invariant.InvariantCalculator(data=self.data)
+        # Set the extrapolation parameters for the low-Q range
+        inv.set_extrapolation(range='low', npts=20, function='guinier')
+        
+        self.assertEqual(inv._low_extrapolation_npts, 20)
+        self.assertEqual(inv._low_extrapolation_function.__class__, invariant.Guinier)
+        
+        # Data boundaries for fiiting
+        qmin = inv._data.x[0]
+        qmax = inv._data.x[inv._low_extrapolation_npts - 1]
+        
+        # Extrapolate the low-Q data
+        a, b = inv._fit(model=inv._low_extrapolation_function,
+                          qmin=qmin,
+                          qmax=qmax,
+                          power=inv._low_extrapolation_power)
+        self.assertAlmostEqual(self.scale, a, 6)
+        self.assertAlmostEqual(self.rg, b, 6)
+        
+        qstar = inv.get_qstar(extrapolation='low')
+        reel_y = self.data.y
+        test_y = inv._low_extrapolation_function.evaluate_model(x=self.data.x)
+        for i in range(len(self.data.x)):
+            value  = math.fabs(test_y[i]-reel_y[i])/reel_y[i]
+            self.assert_(value < 0.001)
+            
+class TestDataExtraLowSlit(unittest.TestCase):
+    """
+        Generate a Guinier distribution and verify that the extrapolation
+        produce the correct ditribution. Tested if the data generated by the 
+        invariant calculator is correct
+    """
+    
+    def setUp(self):
+        """
+            Generate a Guinier distribution. After extrapolating, we will
+            verify that we obtain the scale and rg parameters
+        """
+        list = Loader().load("latex_smeared.xml")
+        self.data = list[0]
+        self.data.dxl = list[0].dxl
+        self.data.dxw = list[0].dxw
+        
+    def test_low_q(self):
+        """
+            Invariant with low-Q extrapolation with slit smear
+        """
+        # Create invariant object. Background and scale left as defaults.
+        inv = invariant.InvariantCalculator(data=self.data)
+        # Set the extrapolation parameters for the low-Q range
+        inv.set_extrapolation(range='low', npts=20, function='guinier')
+        
+        self.assertEqual(inv._low_extrapolation_npts, 20)
+        self.assertEqual(inv._low_extrapolation_function.__class__, invariant.Guinier)
+        
+        # Data boundaries for fiiting
+        qmin = inv._data.x[0]
+        qmax = inv._data.x[inv._low_extrapolation_npts - 1]
+        
+        # Extrapolate the low-Q data
+        a, b = inv._fit(model=inv._low_extrapolation_function,
+                          qmin=qmin,
+                          qmax=qmax,
+                          power=inv._low_extrapolation_power)
+      
+        
+        qstar = inv.get_qstar(extrapolation='low')
+        reel_y = self.data.y
+        #Compution the y 's coming out of the invariant when computing extrapolated
+        #low data . expect the fit engine to have been already called and the guinier
+        # to have the radius and the scale fitted
+        
+        test_y = numpy.zeros(len(self.data.x))
+        smearer = smear_selection(self.data)
+        first_bin, last_bin = smearer.get_bin_range(min(self.data.x),
+                                                         max(self.data.x))
+        test_y[first_bin:last_bin] = inv._low_extrapolation_function.evaluate_model(self.data.x[first_bin:last_bin])
+        test_y = smearer(test_y, first_bin, last_bin) 
+        
+        for i in range(len(self.data.x)):
+            value  = math.fabs(test_y[i]-reel_y[i])/reel_y[i]
+            self.assert_(value < 0.001)
+            
+        # test data coming out of the invariant 
+        test_y = inv._low_extrapolation_function.evaluate_model(x=self.data.x)
+        for i in range(len(self.data.x)):
+            value  = math.fabs(test_y[i]-reel_y[i])/reel_y[i]
+            self.assert_(value < 0.001)
+            
+            
+                    
+