def test_optically_thick_limit_1iso(verbose=True, plot=True, *args, **kwargs):
""" Test that we find Planck in the optically thick limit
In particular, this test will fail if :
- linestrength are not properly calculated
- at noneq, linestrength and emission integrals are mixed up
The test should be run for 1 and several isotopes, because different
calculations paths are used internally, and this can lead to different
errors.
Also, this test is used to run with DEBUG_MODE = True, which will
check that isotopes and molecule ids are what we expect in all the
groupby() loops that make the production code very fast.
Notes
-----
switched from large band calculation with [HITRAN-2016]_ to a calculation with
the embedded [HITEMP-2010]_ fragment (shorter range, but no need to download files)
"""
if plot: # Make sure matplotlib is interactive so that test are not stuck in pytest
plt.ion()
# Force DEBUG_MODE
DEBUG_MODE = radis.DEBUG_MODE
radis.DEBUG_MODE = True
try:
wavenum_min = 2284.2
wavenum_max = 2284.6
P = 0.017 # bar
wstep = 0.001 # cm-1
Tgas = 1200
# %% Generate some CO2 emission spectra
# --------------
sf = SpectrumFactory(
wavenum_min=wavenum_min,
wavenum_max=wavenum_max,
molecule="CO2",
mole_fraction=1,
path_length=0.05,
cutoff=1e-25,
broadening_max_width=1,
export_populations=False, #'vib',
export_lines=False,
isotope=1,
use_cached=True,
wstep=wstep,
pseudo_continuum_threshold=0,
pressure=P,
verbose=False,
)
# sf.fetch_databank('astroquery')
sf.warnings["NegativeEnergiesWarning"] = "ignore"
sf.load_databank("HITEMP-CO2-TEST")
pb = ProgressBar(3, active=verbose)
s_eq = sf.eq_spectrum(Tgas=Tgas, mole_fraction=1, name="Equilibrium")
pb.update(1)
s_2T = sf.non_eq_spectrum(Tvib=Tgas,
Trot=Tgas,
mole_fraction=1,
name="Noneq (2T)")
pb.update(2)
s_4T = sf.non_eq_spectrum(Tvib=(Tgas, Tgas, Tgas),
Trot=Tgas,
mole_fraction=1,
name="Noneq (4T)")
pb.update(3)
s_plck = sPlanck(
wavelength_min=2000, # =wavelength_min,
wavelength_max=
5000, # =wavelength_max - wstep, # there is a border effect on last point
T=Tgas,
)
pb.done()
# %% Post process:
# MAke optically thick, and compare with Planck
for s in [s_eq, s_2T, s_4T]:
s.rescale_path_length(1e6)
if plot:
nfig = "test_opt_thick_limit_1iso {0}".format(s.name)
plt.figure(nfig).clear()
s.plot(wunit="nm", nfig=nfig, lw=4)
s_plck.plot(wunit="nm", nfig=nfig, Iunit="mW/cm2/sr/nm", lw=2)
plt.legend()
if verbose:
printm(
"Residual between opt. thick CO2 spectrum ({0}) and Planck: {1:.2g}"
.format(
s.name,
get_residual(s,
s_plck,
"radiance_noslit",
ignore_nan=True),
))
# assert get_residual(s, s_plck, 'radiance_noslit', ignore_nan=True) < 1e-3
assert get_residual(s, s_plck, "radiance_noslit",
ignore_nan=True) < 0.9e-4
if verbose:
printm("Tested optically thick limit is Planck (1 isotope): OK")
finally:
# Reset DEBUG_MODE
radis.DEBUG_MODE = DEBUG_MODE
verbose=False,
broadening_max_width=broadening_max_width,
wstep=wstep,
warnings={'MissingSelfBroadeningWarning': 'ignore'},
export_lines=False,
chunksize='DLM',
)
sf.fetch_databank(
load_energies=False) # loads from HITRAN, requires an internet connection
#%% ===========================================================================
# Calculations
# =============================================================================
# Calculate ground emission
s_earth_0 = sPlanck(wmin, wmax, wstep=wstep, T=T_earth, eps=1 - albedo)
#%% Calculate atmosphere
# Calculate atmosphere layers
slabs = []
print('Calculating Atmosphere layers')
pb = ProgressBar(len(atm))
for i, r in atm.iterrows():
pb.update(i)
s = sf.eq_spectrum(
Tgas=r.T_K,
mole_fraction=x_CO2,
path_length=r.path_length * 1e5, # cm
pressure=r.P_Pa * 1e-5, # bar
