source: sasview/src/sas/qtgui/Perspectives/Fitting/UnitTesting/FittingUtilitiesTest.py @ 6b43c58

Last change on this file since 6b43c58 was 06b0138, checked in by Piotr Rozyczko <rozyczko@…>, 8 years ago

Polydispersity support: cleanup and unit tests SASVIEW-575
  • Property mode set to 100755
File size: 9.7 KB
Line 
1import sys
2import unittest
3from PyQt4 import QtGui
4
5from sas.qtgui.Plotting.PlotterData import Data1D
6from sas.qtgui.Plotting.PlotterData import Data2D
7
8from UnitTesting.TestUtils import WarningTestNotImplemented
9
10from sasmodels import generate
11from sasmodels import modelinfo
12from sasmodels.sasview_model import load_standard_models
13
14# Tested module
15from sas.qtgui.Perspectives.Fitting import FittingUtilities
16
17class FittingUtilitiesTest(unittest.TestCase):
18    '''Test the Fitting Utilities functions'''
19    def setUp(self):
20        '''Empty'''
21        pass
22
23    def tearDown(self):
24        '''Empty'''
25        pass
26
27    def testReplaceShellName(self):
28        """
29        Test the utility function for string manipulation
30        """
31        param_name = "test [123]"
32        value = "replaced"
33        result = FittingUtilities.replaceShellName(param_name, value)
34       
35        self.assertEqual(result, "test replaced")
36
37        # Assert!
38        param_name = "no brackets"
39        with self.assertRaises(AssertionError):
40            result = FittingUtilities.replaceShellName(param_name, value)
41
42       
43    def testGetIterParams(self):
44        """
45        Assure the right multishell parameters are returned
46        """
47        # Use a single-shell parameter
48        model_name = "barbell"
49        kernel_module = generate.load_kernel_module(model_name)
50        barbell_parameters = modelinfo.make_parameter_table(getattr(kernel_module, 'parameters', []))
51
52        params = FittingUtilities.getIterParams(barbell_parameters)
53        # returns empty list
54        self.assertEqual(params, [])
55
56        # Use a multi-shell parameter
57        model_name = "core_multi_shell"
58        kernel_module = generate.load_kernel_module(model_name)
59        multishell_parameters = modelinfo.make_parameter_table(getattr(kernel_module, 'parameters', []))
60
61        params = FittingUtilities.getIterParams(multishell_parameters)
62        # returns a non-empty list
63        self.assertNotEqual(params, [])
64        self.assertIn('sld', str(params))
65        self.assertIn('thickness', str(params))
66
67    def testGetMultiplicity(self):
68        """
69        Assure more multishell parameters are evaluated correctly
70        """
71        # Use a single-shell parameter
72        model_name = "barbell"
73        kernel_module = generate.load_kernel_module(model_name)
74        barbell_parameters = modelinfo.make_parameter_table(getattr(kernel_module, 'parameters', []))
75
76        param_name, param_length = FittingUtilities.getMultiplicity(barbell_parameters)
77        # returns nothing
78        self.assertEqual(param_name, "")
79        self.assertEqual(param_length, 0)
80
81        # Use a multi-shell parameter
82        model_name = "core_multi_shell"
83        kernel_module = generate.load_kernel_module(model_name)
84        multishell_parameters = modelinfo.make_parameter_table(getattr(kernel_module, 'parameters', []))
85
86        param_name, param_length = FittingUtilities.getMultiplicity(multishell_parameters)
87
88        self.assertEqual(param_name, "n")
89        self.assertEqual(param_length, 10)
90
91    def testAddParametersToModel(self):
92        """
93        Checks the QModel update from Sasmodel parameters
94        """
95        # Use a single-shell parameter
96        model_name = "barbell"
97        models = load_standard_models()
98
99        kernel_module = generate.load_kernel_module(model_name)
100        kernel_module_o = None
101        for model in models:
102            if model.name == model_name:
103                kernel_module_o = model()
104        self.assertIsNotNone(kernel_module_o)
105        barbell_parameters = modelinfo.make_parameter_table(getattr(kernel_module, 'parameters', []))
106
107        params = FittingUtilities.addParametersToModel(barbell_parameters, kernel_module_o, True)
108
109        # Test the resulting model
110        self.assertEqual(len(params), 7)
111        self.assertEqual(len(params[0]), 5)
112        self.assertTrue(params[0][0].isCheckable())
113        self.assertEqual(params[0][0].text(), "sld")
114        self.assertEqual(params[1][0].text(), "sld_solvent")
115
116        # Use a multi-shell parameter to see that the method includes shell params
117        model_name = "core_multi_shell"
118        kernel_module = generate.load_kernel_module(model_name)
119        kernel_module_o = None
120        for model in models:
121            if model.name == model_name:
122                kernel_module_o = model()
123        self.assertIsNotNone(kernel_module_o)
124        multi_parameters = modelinfo.make_parameter_table(getattr(kernel_module, 'parameters', []))
125
126        params = FittingUtilities.addParametersToModel(multi_parameters, kernel_module_o, False)
127
128        # Test the resulting model
129        self.assertEqual(len(params), 3)
130        self.assertEqual(len(params[0]), 5)
131        self.assertTrue(params[0][0].isCheckable())
132        self.assertEqual(params[0][0].text(), "sld_core")
133        self.assertEqual(params[1][0].text(), "radius")
134
135    def testAddSimpleParametersToModel(self):
136        """
137        Checks the QModel update from Sasmodel parameters - no polydisp
138        """
139        # Use a multi-shell parameter to see that the method doesn't include shells
140        model_name = "core_multi_shell"
141        kernel_module = generate.load_kernel_module(model_name)
142        models = load_standard_models()
143        kernel_module_o = None
144        for model in models:
145            if model.name == model_name:
146                kernel_module_o = model()
147        self.assertIsNotNone(kernel_module_o)
148        multi_parameters = modelinfo.make_parameter_table(getattr(kernel_module, 'parameters', []))
149
150        params = FittingUtilities.addParametersToModel(multi_parameters, kernel_module_o, True)
151
152        # Test the resulting model
153        self.assertEqual(len(params), 3)
154        self.assertEqual(len(params[0]), 5)
155        self.assertTrue(params[0][0].isCheckable())
156        self.assertEqual(params[0][0].text(), "sld_core")
157        self.assertEqual(params[1][0].text(), "radius")
158
159    def testAddCheckedListToModel(self):
160        """
161        Test for inserting a checkboxed item into a QModel
162        """
163        model = QtGui.QStandardItemModel()
164        params = ["row1", "row2", "row3"]
165
166        FittingUtilities.addCheckedListToModel(model, params)
167
168        # Check the model
169        self.assertEqual(model.rowCount(), 1)
170        self.assertTrue(model.item(0).isCheckable())
171        self.assertEqual(model.item(0, 0).text(), params[0])
172        self.assertEqual(model.item(0, 1).text(), params[1])
173        self.assertEqual(model.item(0, 2).text(), params[2])
174
175    def testAddShellsToModel(self):
176        """
177        Test for inserting a list of QItems into a model
178        """
179        # Use a multi-shell parameter to see that the method doesn't include shells
180        model_name = "core_multi_shell"
181        kernel_module = generate.load_kernel_module(model_name)
182        multi_parameters = modelinfo.make_parameter_table(getattr(kernel_module, 'parameters', []))
183
184        model = QtGui.QStandardItemModel()
185
186        index = 2
187        FittingUtilities.addShellsToModel(multi_parameters, model, index)
188        # There should be index*len(multi_parameters) new rows
189        self.assertEqual(model.rowCount(), 4)
190
191        model = QtGui.QStandardItemModel()
192        index = 5
193        FittingUtilities.addShellsToModel(multi_parameters, model, index)
194        self.assertEqual(model.rowCount(), 10)
195       
196        self.assertEqual(model.item(1).child(0).text(), "Polydispersity")
197        self.assertEqual(model.item(1).child(0).child(0).text(), "Distribution")
198        self.assertEqual(model.item(1).child(0).child(0,1).text(), "40.0")
199
200    def testCalculate1DChi2(self):
201        """
202        Test the chi2 calculator for Data1D
203        """
204        reference_data = Data1D(x=[0.1, 0.2], y=[0.0, 0.0])
205
206        # 1. identical data
207        current_data = Data1D(x=[0.1, 0.2], y=[0.0, 0.0])
208
209        chi = FittingUtilities.calculateChi2(reference_data, current_data)
210
211        # Should be zero
212        self.assertAlmostEqual(chi, 0.0, 8)
213
214        # 2. far data
215        current_data = Data1D(x=[0.1, 0.2], y=[200.0, 150.0])
216
217        chi = FittingUtilities.calculateChi2(reference_data, current_data)
218
219        # Should not be zero
220        self.assertAlmostEqual(chi, 31250.0, 8)
221
222        # 3. Wrong data
223        current_data = Data1D(x=[0.1, 0.2], y=[200.0, 150.0, 200.0])
224        chi = FittingUtilities.calculateChi2(reference_data, current_data)
225        # Should be None
226        self.assertIsNone(chi)
227
228    def testCalculate2DChi2(self):
229        """
230        Test the chi2 calculator for Data2D
231        """
232        reference_data = Data2D(image=[1.0, 2.0, 3.0],
233                      err_image=[0.01, 0.02, 0.03],
234                      qx_data=[0.1, 0.2, 0.3],
235                      qy_data=[0.1, 0.2, 0.3])
236
237        # 1. identical data
238        current_data = Data2D(image=[1.0, 2.0, 3.0],
239                      err_image=[0.01, 0.02, 0.03],
240                      qx_data=[0.1, 0.2, 0.3],
241                      qy_data=[0.1, 0.2, 0.3])
242
243        chi = FittingUtilities.calculateChi2(reference_data, current_data)
244
245        # Should be zero
246        self.assertAlmostEqual(chi, 0.0, 8)
247
248        # 2. far data
249        current_data = Data2D(image=[100.0, 200.0, 300.0],
250                      err_image=[1.01, 2.02, 3.03],
251                      qx_data=[0.1, 0.2, 0.3],
252                      qy_data=[100.0, 200., 300.])
253
254        chi = FittingUtilities.calculateChi2(reference_data, current_data)
255
256        # Should not be zero
257        self.assertAlmostEqual(chi, 9607.88, 2)
258
259        # 3. Wrong data
260        current_data = Data2D(image=[1.0, 2.0, 3.0],
261                      err_image=[0.01, 0.02],
262                      qx_data=[0.1, 0.2],
263                      qy_data=[0.1, 0.2, 0.3])
264        # Should throw
265        with self.assertRaises(ValueError):
266            chi = FittingUtilities.calculateChi2(reference_data, current_data)
267
268if __name__ == "__main__":
269    unittest.main()
Note: See TracBrowser for help on using the repository browser.