1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67
|
import unittest
import xraylib
class TestComptonProfiles(unittest.TestCase):
def test_pz_0(self):
profile = xraylib.ComptonProfile(26, 0.0)
self.assertAlmostEqual(profile, 7.060, delta=1E-6)
profile = xraylib.ComptonProfile_Partial(26, xraylib.N1_SHELL, 0.0)
self.assertAlmostEqual(profile, 1.550, delta=1E-6)
profile1 = xraylib.ComptonProfile_Partial(26, xraylib.L2_SHELL, 0.0)
profile2 = xraylib.ComptonProfile_Partial(26, xraylib.L3_SHELL, 0.0)
self.assertAlmostEqual(profile1, profile2, delta=1E-6)
self.assertAlmostEqual(profile1, 0.065, delta=1E-6)
def test_pz_100(self):
profile = xraylib.ComptonProfile(26, 100.0)
self.assertAlmostEqual(profile, 1.8E-5, delta=1E-8)
profile = xraylib.ComptonProfile_Partial(26, xraylib.N1_SHELL, 100.0)
self.assertAlmostEqual(profile, 5.1E-9, delta=1E-12)
profile1 = xraylib.ComptonProfile_Partial(26, xraylib.L2_SHELL, 100.0)
profile2 = xraylib.ComptonProfile_Partial(26, xraylib.L3_SHELL, 100.0)
self.assertAlmostEqual(profile1, profile2, delta=1E-10)
self.assertAlmostEqual(profile1, 1.1E-8, delta=1E-10)
def test_pz_50(self):
profile = xraylib.ComptonProfile(26, 50.0)
self.assertAlmostEqual(profile, 0.0006843950273082384, delta=1E-8)
profile = xraylib.ComptonProfile_Partial(26, xraylib.N1_SHELL, 50.0)
self.assertAlmostEqual(profile, 2.4322755767709126e-07, delta=1E-10)
profile1 = xraylib.ComptonProfile_Partial(26, xraylib.L2_SHELL, 50.0)
profile2 = xraylib.ComptonProfile_Partial(26, xraylib.L3_SHELL, 50.0)
self.assertAlmostEqual(profile1, profile2, delta=1E-10)
self.assertAlmostEqual(profile1, 2.026953933016568e-06, delta=1E-10)
def test_bad_input(self):
with self.assertRaises(ValueError):
xraylib.ComptonProfile(0, 0.0)
xraylib.ComptonProfile(102, 0.0)
with self.assertRaises(ValueError):
xraylib.ComptonProfile(103, 0.0)
with self.assertRaises(ValueError):
xraylib.ComptonProfile(26, -1.0)
with self.assertRaises(ValueError):
xraylib.ComptonProfile_Partial(0, xraylib.K_SHELL, 0.0)
xraylib.ComptonProfile_Partial(102, xraylib.K_SHELL, 0.0)
with self.assertRaises(ValueError):
xraylib.ComptonProfile_Partial(103, xraylib.K_SHELL, 0.0)
with self.assertRaises(ValueError):
xraylib.ComptonProfile_Partial(26, xraylib.K_SHELL, -1.0)
with self.assertRaises(ValueError):
xraylib.ComptonProfile_Partial(26, -1, 0.0)
with self.assertRaises(ValueError):
xraylib.ComptonProfile_Partial(26, xraylib.N2_SHELL, 0.0)
with self.assertRaises(TypeError):
xraylib.ComptonProfile_Partial()
with self.assertRaises(TypeError):
xraylib.ComptonProfile_Partial("26", xraylib.N2_SHELL, 0.0)
if __name__ == '__main__':
unittest.main(verbosity=2)
|