def test_read_zeta_data(default_atom_h5_path, chianti_he_db_h5_path): data = atomic.read_zeta_data(chianti_he_db_h5_path) testing.assert_almost_equal(data[2000][1][1], 0.339, decimal=4) testing.assert_almost_equal(data[2000][1][2], 0.000, decimal=4) with pytest.raises(ValueError): atomic.read_zeta_data(None) with pytest.raises(IOError): atomic.read_zeta_data('fakepath') with pytest.raises(ValueError): atomic.read_zeta_data(default_atom_h5_path)
def test_read_collision_data(default_atom_h5_path, chianti_he_db_h5_path): data = atomic.read_collision_data(chianti_he_db_h5_path) assert data[0]['atomic_number'][0] == 2 assert data[0]['ion_number'][0] == 0 assert data[0]['level_number_upper'][0] == 18 assert data[0]['level_number_lower'][0] == 2 assert data[0]['g_ratio'][0] == 1.0 testing.assert_almost_equal(data[0]['delta_e'][0], 35484.251143, decimal=4) assert data[1][0] == 2000.0 assert data[1][1] == 4000.0 with pytest.raises(ValueError): atomic.read_zeta_data(None) with pytest.raises(IOError): atomic.read_zeta_data('fakepath') with pytest.raises(ValueError): atomic.read_zeta_data(default_atom_h5_path)