def test_he_nlte_plasma(number_density, atomic_data, time_explosion, t_rad, w, j_blues): he_nlte_plasma = LegacyPlasmaArray(number_densities=number_density, atomic_data=atomic_data, time_explosion=time_explosion, helium_treatment='recomb-nlte') he_nlte_plasma.update_radiationfield(t_rad, w, j_blues, nlte_config=None) assert np.allclose(he_nlte_plasma.get_value( 'ion_number_density').ix[2].ix[1], number_density.ix[2]) assert np.all(he_nlte_plasma.get_value( 'level_number_density').ix[2].ix[0].ix[0])==0.0 assert np.allclose(he_nlte_plasma.get_value( 'level_number_density').ix[2].sum(), number_density.ix[2])
def test_he_nlte_plasma(number_density, atomic_data, time_explosion, t_rad, w, j_blues): he_nlte_plasma = LegacyPlasmaArray(number_densities=number_density, atomic_data=atomic_data, time_explosion=time_explosion, helium_treatment='recomb-nlte') he_nlte_plasma.update_radiationfield(t_rad, w, j_blues, nlte_config=None) assert np.allclose( he_nlte_plasma.get_value('ion_number_density').ix[2].ix[1], number_density.ix[2]) assert np.all( he_nlte_plasma.get_value( 'level_number_density').ix[2].ix[0].ix[0]) == 0.0 assert np.allclose( he_nlte_plasma.get_value('level_number_density').ix[2].sum(), number_density.ix[2])