class TestCoreShellCylinder(unittest.TestCase): """ Unit tests for calculate_ER (CoreShellcylinder model) """ def setUp(self): from sas.models.CoreShellCylinderModel import CoreShellCylinderModel from sas.models.DiamCylFunc import DiamCylFunc self.comp = CoreShellCylinderModel() self.diam = DiamCylFunc() def test(self): """ Test 1D model for a CoreShellCylinder """ self.comp.setParam("radius", 20) self.comp.setParam("thickness", 10) self.comp.setParam("length", 400) self.diam.setParam("radius", 30) self.diam.setParam("length", 420) self.assertAlmostEqual(self.comp.calculate_ER(), self.diam.run(0.1) / 2)
class TestCoreShellCylinder(unittest.TestCase): """ Unit tests for calculate_ER (CoreShellcylinder model) """ def setUp(self): from sas.models.CoreShellCylinderModel import CoreShellCylinderModel from sas.models.DiamCylFunc import DiamCylFunc self.comp = CoreShellCylinderModel() self.diam = DiamCylFunc() def test(self): """ Test 1D model for a CoreShellCylinder """ self.comp.setParam("radius", 20) self.comp.setParam("thickness", 10) self.comp.setParam("length",400) self.diam.setParam("radius", 30) self.diam.setParam("length",420) self.assertAlmostEqual(self.comp.calculate_ER(), self.diam.run(0.1)/2)
class TestCoreShellCylinder(unittest.TestCase): """ Testing C++ Cylinder model """ def setUp(self): from sas.models.CoreShellCylinderModel import CoreShellCylinderModel self.model= CoreShellCylinderModel() self.model.setParam('scale', 1.0) self.model.setParam('radius', 20.0) self.model.setParam('thickness', 10.0) self.model.setParam('length', 400.0) self.model.setParam('core_sld', 1.e-6) self.model.setParam('shell_sld', 4.e-6) self.model.setParam('solvent_sld', 1.e-6) self.model.setParam('background', 0.0) self.model.setParam('axis_theta', 0.0) self.model.setParam('axis_phi', 90.0) def test_simple(self): """ Test simple 1D and 2D values Numbers taken from model that passed validation, before the update to C++ underlying class. """ self.assertAlmostEqual(self.model.run(0.001), 353.55013216754583, 3) self.assertAlmostEqual(self.model.runXY([0.001,0.001]), 355.25355270620543, 3) def test_dispersion(self): """ Test with dispersion """ from sas.models.DisperseModel import DisperseModel disp = DisperseModel(self.model, ['radius', 'thickness', 'length'], [5, 2, 50]) disp.setParam('n_pts', 10) self.assertAlmostEqual(disp.run(0.001), 358.44062724936009, 3) self.assertAlmostEqual(disp.runXY([0.001,0.001]), 360.22673635224584, 3) def test_new_disp(self): from sas.models.dispersion_models import GaussianDispersion disp_rm = GaussianDispersion() self.model.set_dispersion('radius', disp_rm) self.model.dispersion['radius']['width'] = 0.25 self.model.dispersion['radius']['npts'] = 10 self.model.dispersion['radius']['nsigmas'] = 2 disp_rr = GaussianDispersion() self.model.set_dispersion('thickness', disp_rr) self.model.dispersion['thickness']['width'] = 0.2 self.model.dispersion['thickness']['npts'] = 10 self.model.dispersion['thickness']['nsigmas'] = 2 disp_len = GaussianDispersion() self.model.set_dispersion('length', disp_len) self.model.dispersion['length']['width'] = 1.0/8.0 self.model.dispersion['length']['npts'] = 10 self.model.dispersion['length']['nsigmas'] = 2 self.assertAlmostEqual(self.model.run(0.001), 1.07832610*358.44062724936009, 3) self.assertAlmostEqual(self.model.runXY([0.001,0.001]), 1.07844010*360.22673635224584, 3) def test_array(self): """ Perform complete rotational average and compare to 1D """ from sas.models.dispersion_models import ArrayDispersion disp_ph = ArrayDispersion() disp_th = ArrayDispersion() values_ph = numpy.zeros(100) values_th = numpy.zeros(100) weights = numpy.zeros(100) for i in range(100): values_ph[i]=(360/99.0*i) values_th[i]=(180/99.0*i) weights[i]=(1.0) disp_ph.set_weights(values_ph, weights) disp_th.set_weights(values_th, weights) self.model.set_dispersion('axis_theta', disp_th) self.model.set_dispersion('axis_phi', disp_ph) val_1d = self.model.run(math.sqrt(0.0002)) val_2d = self.model.runXY([0.01,0.01]) self.assertTrue(math.fabs(val_1d-val_2d)/val_1d < 0.02)