source: sasview/test/sascalculator/test/utest_sas_gen.py @ bc4f5bb

magnetic_scattrelease-4.2.2ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249ticket885unittest-saveload
Last change on this file since bc4f5bb was f926abb, checked in by Paul Kienzle <pkienzle@…>, 6 years ago

add smoke test for generic scattering calculator

  • Property mode set to 100644
File size: 1.7 KB
Line 
1"""
2Unit tests for the sas_gen
3"""
4import warnings
5warnings.simplefilter("ignore")
6
7import unittest
8import numpy as np
9
10from sas.sascalc.calculator import sas_gen
11
12
13class sas_gen_test(unittest.TestCase):
14
15    def setUp(self):
16        self.sldloader = sas_gen.SLDReader()
17        self.pdbloader = sas_gen.PDBReader()
18        self.omfloader = sas_gen.OMFReader()
19
20    def test_sldreader(self):
21        """
22        Test .sld file loaded
23        """
24        f = self.sldloader.read("sld_file.sld")
25        self.assertEqual(f.pos_x[0], -40.5)
26        self.assertEqual(f.pos_y[0], -13.5)
27        self.assertEqual(f.pos_z[0], -13.5)
28
29    def test_pdbreader(self):
30        """
31        Test .pdb file loaded
32        """
33        f = self.pdbloader.read("c60.pdb")
34        self.assertEqual(f.pos_x[0], -0.733)
35        self.assertEqual(f.pos_y[0], -1.008)
36        self.assertEqual(f.pos_z[0], 3.326)
37
38    def test_omfreader(self):
39        """
40        Test .omf file loaded
41        """
42        f = self.omfloader.read("A_Raw_Example-1.omf")
43        output = sas_gen.OMF2SLD()
44        output.set_data(f)
45        self.assertEqual(f.mx[0], 0)
46        self.assertEqual(f.my[0], 0)
47        self.assertEqual(f.mz[0], 0)
48        self.assertEqual(output.pos_x[0], 0.0)
49        self.assertEqual(output.pos_y[0], 0.0)
50        self.assertEqual(output.pos_z[0], 0.0)
51
52    def test_calculator(self):
53        """
54        Test that the calculator calculates.
55        """
56        f = self.omfloader.read("A_Raw_Example-1.omf")
57        omf2sld = sas_gen.OMF2SLD()
58        omf2sld.set_data(f)
59        model = sas_gen.GenSAS()
60        model.set_sld_data(omf2sld.output)
61        x = np.linspace(0, 0.1, 11)[1:]
62        model.runXY([x, x])
63
64
65if __name__ == '__main__':
66    unittest.main()
67
Note: See TracBrowser for help on using the repository browser.