def test_el_z_inputs_same(): el_ioneq_input = ioneq(el) z_ioneq_input = ioneq(z) el_ioneq_input.load() z_ioneq_input.load() assert np.all(el_ioneq_input.Temperature == z_ioneq_input.Temperature) assert np.all(el_ioneq_input.Ioneq == z_ioneq_input.Ioneq)
def test_el_z_inputs_same(): el_ioneq_input = ioneq(el) z_ioneq_input = ioneq(z) el_ioneq_input.load() z_ioneq_input.load() assert np.all(el_ioneq_input.Temperature == z_ioneq_input.Temperature) assert np.all(el_ioneq_input.Ioneq == z_ioneq_input.Ioneq)
def test_equlibrium_state_vs_chiantipy(natom=8): """ Test equilibrium states saved in EigenData2 and compare them with Outputs from ChiantiPy. Note: This test requires ChiantiPy to be installed (see details in: https://github.com/chianti-atomic/ChiantiPy). """ try: import ChiantiPy.core as ch except ImportError: warnings.warn('ChiantiPy is required in this test.', UserWarning) return temperatures = [1.0e4, 1.0e5, 1.0e6, 1.0e7, 1.0e8] eqi_ch = ch.ioneq(natom) eqi_ch.calculate(temperatures) conce = eqi_ch.Ioneq table_sta = nei.EigenData2(element=natom) for i in range(2): ch_conce = conce[:, i] table_conce = table_sta.equilibrium_state(T_e=temperatures[i]) assert ch_conce.all() == table_conce.all() return
import ChiantiPy.core as ch import numpy as np import ChiantiPy.tools.filters as chfilters import matplotlib.pyplot as plt t = 10.**(3.84 + 0.05 * np.arange(8.)) H1 = ch.ion('h_1', temperature=t, eDensity=1.e+14, em=1.e+27) H1.popPlot(top=10) H1.intensityPlot(wvlRange=[6564, 6565], linLog='log', index=4, top=10) H1.intensityList(wvlRange=[4000, 7000], index=4, top=10) plt.show(H1) H1a = ch.ioneq(1) H1a.load() H1a.plot() #plt.xlim(0,3e4) plt.tight_layout() plt.show(H1a)
def test_load_ioneq_alternate_file(): load_ioneq = ioneq(el) load_ioneq.load(ioneqName='chianti') assert hasattr(load_ioneq, 'Temperature') assert hasattr(load_ioneq, 'Ioneq')
def test_load_ioneq(): load_ioneq = ioneq(el) load_ioneq.load() assert hasattr(load_ioneq, 'Temperature') assert hasattr(load_ioneq, 'Ioneq')
def test_calculate_ioneq(): calc_ioneq = ioneq(el) calc_ioneq.calculate(temperature) assert hasattr(calc_ioneq, 'Temperature') assert hasattr(calc_ioneq, 'Ioneq') assert np.all(calc_ioneq.Temperature == temperature)
def test_z_input(): z_ioneq_input = ioneq(z)
def test_el_input(): el_ioneq_input = ioneq(el) assert el_ioneq_input.Z == util.el2z(el)
def test_load_ioneq_alternate_file(): load_ioneq = ioneq(el) load_ioneq.load(ioneqName='chianti') assert hasattr(load_ioneq, 'Temperature') assert hasattr(load_ioneq, 'Ioneq')
def test_load_ioneq(): load_ioneq = ioneq(el) load_ioneq.load() assert hasattr(load_ioneq, 'Temperature') assert hasattr(load_ioneq, 'Ioneq')
def test_calculate_ioneq(): calc_ioneq = ioneq(el) calc_ioneq.calculate(temperature) assert hasattr(calc_ioneq, 'Temperature') assert hasattr(calc_ioneq, 'Ioneq') assert np.all(calc_ioneq.Temperature == temperature)
def test_z_input(): z_ioneq_input = ioneq(z)
def test_el_input(): el_ioneq_input = ioneq(el) assert el_ioneq_input.Z == util.el2z(el)