diff --git a/helita/sim/tests/test_rh15d.py b/helita/sim/tests/test_rh15d.py new file mode 100644 index 00000000..fbf1d7eb --- /dev/null +++ b/helita/sim/tests/test_rh15d.py @@ -0,0 +1,90 @@ +# -*- coding: utf-8 -*- +""" +Tests for the rh15d module +""" + +import numpy as np +from helita.sim import rh15d + +TMP_ATOM_FILENAME = 'atom.tmp' +TEST_LEVELS = [ + " 0.000 2.00 'CA II 3P6 4S 2SE ' 1 0", + " 13650.190 4.00 'CA II 3P6 3D 2DE 3 ' 1 1", + " 13710.880 6.00 'CA II 3P6 3D 2DE 5 ' 1 2", + " 25191.510 2.00 'CA II 3P6 4P 2PO 1 ' 1 3", + " 25414.400 4.00 'CA II 3P6 4P 2PO 3 ' 1 4", + " 95785.470 1.00 'CA III 3P6 1SE ' 2 5" +] + +TEST_LEVELS_DATA = np.array([( 0. , 2., 'CA II 3P6 4S 2SE', 1, 0), + (13650.19, 4., 'CA II 3P6 3D 2DE 3', 1, 1), + (13710.88, 6., 'CA II 3P6 3D 2DE 5', 1, 2), + (25191.51, 2., 'CA II 3P6 4P 2PO 1', 1, 3), + (25414.4 , 4., 'CA II 3P6 4P 2PO 3', 1, 4), + (95785.47, 1., 'CA III 3P6 1SE', 2, 5)], + dtype=[('energy', '