Changeset e84e1e6 in sasview for DataLoader/test/utest_averaging.py

Timestamp:

Jan 26, 2010 8:43:38 PM (15 years ago)

Author:

Mathieu Doucet <doucetm@…>

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:

c1469ebe

Parents:

25e3fc9

Message:

dataloader: added unit tests to start systematically checking sector averaging computations.

File:

• DataLoader/test/utest_averaging.py (modified) (1 diff)

DataLoader/test/utest_averaging.py

@@ -3 +3 @@
 
 from DataLoader.loader import  Loader
-from DataLoader.manipulations import Ring, CircularAverage
+from DataLoader.manipulations import Ring, CircularAverage, SectorPhi, get_q
  
 import os.path
+import numpy, math
+import DataLoader.data_info as data_info
+
+class Averaging(unittest.TestCase):
+    """
+        Test averaging manipulations on a flat distribution
+    """
+    def setUp(self):
+        """
+            Create a flat 2D distribution. All averaging results
+            should return the predefined height of the distribution (1.0).
+        """
+        x_0  = numpy.ones([100,100])
+        dx_0 = numpy.ones([100,100])
+        self.data = data_info.Data2D(x_0, dx_0)
+        detector = data_info.Detector()
+        detector.distance = 1000.0
+        detector.pixel_size.x = 1.0
+        detector.pixel_size.y = 1.0
+        detector.beam_center.x = 50
+        detector.beam_center.y = 50
+        self.data.detector.append(detector)
+        
+        source = data_info.Source()
+        source.wavelength = 10.0
+        self.data.source = source
+        
+        self.qmin = get_q(1.0, 1.0, 1000.0, 10.0)
+        
+    def test_ring_flat_distribution(self):
+        """
+            Test ring averaging
+        """
+        r = Ring(r_min=2*self.qmin, r_max=5*self.qmin, 
+                 center_x=self.data.detector[0].beam_center.x, 
+                 center_y=self.data.detector[0].beam_center.y)
+        r.nbins_phi = 20
+        
+        o = r(self.data)
+        for i in range(20):
+            self.assertEqual(o.y[i], 1.0)
+            
+    def test_sectorphi_full(self):
+        """
+            Test sector averaging
+        """
+        r = SectorPhi(r_min=self.qmin, r_max=3*self.qmin, phi_min=0, phi_max=math.pi*2.0)
+        r.nbins_phi = 20
+        o = r(self.data)
+        for i in range(20):
+            self.assertEqual(o.y[i], 1.0)
+            
+            
+    def test_sectorphi_partial(self):
+        """
+        """
+        phi_max = 1.5
+        r = SectorPhi(r_min=self.qmin, r_max=3*self.qmin, phi_min=0, phi_max=phi_max)
+        self.assertEqual(r.phi_max, phi_max)
+        r.nbins_phi = 20
+        o = r(self.data)
+        self.assertEqual(r.phi_max, phi_max)
+        for i in range(20):
+            self.assertEqual(o.y[i], 1.0)
+            
+            
 
 class data_info_tests(unittest.TestCase):