[959eb01] | 1 | """ |
---|
| 2 | Unit tests for the sas_gen |
---|
| 3 | """ |
---|
| 4 | import warnings |
---|
| 5 | warnings.simplefilter("ignore") |
---|
| 6 | |
---|
| 7 | import unittest |
---|
| 8 | from sas.sascalc.calculator import sas_gen |
---|
| 9 | |
---|
| 10 | |
---|
| 11 | class sas_gen_test(unittest.TestCase): |
---|
[1d014cb] | 12 | |
---|
[959eb01] | 13 | def setUp(self): |
---|
| 14 | self.sldloader = sas_gen.SLDReader() |
---|
| 15 | self.pdbloader = sas_gen.PDBReader() |
---|
| 16 | self.omfloader = sas_gen.OMFReader() |
---|
[1d014cb] | 17 | |
---|
[959eb01] | 18 | def test_sldreader(self): |
---|
| 19 | """ |
---|
[1d014cb] | 20 | Test .sld file loaded |
---|
[959eb01] | 21 | """ |
---|
| 22 | f = self.sldloader.read("sld_file.sld") |
---|
| 23 | self.assertEqual(f.pos_x[0], -40.5) |
---|
| 24 | self.assertEqual(f.pos_y[0], -13.5) |
---|
| 25 | self.assertEqual(f.pos_z[0], -13.5) |
---|
[1d014cb] | 26 | |
---|
[959eb01] | 27 | def test_pdbreader(self): |
---|
| 28 | """ |
---|
[1d014cb] | 29 | Test .pdb file loaded |
---|
[959eb01] | 30 | """ |
---|
| 31 | f = self.pdbloader.read("c60.pdb") |
---|
| 32 | self.assertEqual(f.pos_x[0], -0.733) |
---|
| 33 | self.assertEqual(f.pos_y[0], -1.008) |
---|
| 34 | self.assertEqual(f.pos_z[0], 3.326) |
---|
[1d014cb] | 35 | |
---|
[959eb01] | 36 | def test_omfreader(self): |
---|
| 37 | """ |
---|
[1d014cb] | 38 | Test .omf file loaded |
---|
[959eb01] | 39 | """ |
---|
| 40 | f = self.omfloader.read("A_Raw_Example-1.omf") |
---|
| 41 | output = sas_gen.OMF2SLD() |
---|
| 42 | output.set_data(f) |
---|
| 43 | self.assertEqual(f.mx[0], 0) |
---|
| 44 | self.assertEqual(f.my[0], 0) |
---|
| 45 | self.assertEqual(f.mz[0], 0) |
---|
| 46 | self.assertEqual(output.pos_x[0], 0.0) |
---|
| 47 | self.assertEqual(output.pos_y[0], 0.0) |
---|
| 48 | self.assertEqual(output.pos_z[0], 0.0) |
---|
| 49 | |
---|
| 50 | |
---|
| 51 | if __name__ == '__main__': |
---|
| 52 | unittest.main() |
---|
| 53 | |
---|