import unittest import xraylib class TestAtomicLevelWidth(unittest.TestCase): def test_Fe_K(self): width = xraylib.AtomicLevelWidth(26, xraylib.K_SHELL) self.assertAlmostEqual(width, 1.19E-3) def test_U_N7(self): width = xraylib.AtomicLevelWidth(92, xraylib.N7_SHELL) self.assertAlmostEqual(width, 0.31E-3) def test_bad_Z(self): with self.assertRaises(ValueError): width = xraylib.AtomicLevelWidth(185, xraylib.K_SHELL) def test_bad_shell(self): with self.assertRaises(ValueError): width = xraylib.AtomicLevelWidth(26, -5) def test_invalid_shell(self): with self.assertRaises(ValueError): width = xraylib.AtomicLevelWidth(26, xraylib.N3_SHELL) if __name__ == '__main__': unittest.main(verbosity=2)