def test_irrelevant_file_loading(*args, **kwargs):
""" check that irrelevant files (irrelevant wavenumber) are not loaded """
# For cdsd-hitemp files :
# Ensures file is not empty
df = cdsd2df(getTestFile("cdsd_hitemp_09_header.txt"), cache="regen")
assert len(df) > 0
# Now test nothing is loaded if asking for outside the wavenumber range
with pytest.raises(IrrelevantFileWarning):
df = cdsd2df(getTestFile("cdsd_hitemp_09_header.txt"),
load_wavenum_min=100000)
with pytest.raises(IrrelevantFileWarning):
df = cdsd2df(getTestFile("cdsd_hitemp_09_header.txt"),
load_wavenum_max=0.5)
# For HITRAN files :
# Ensures file is not empty
df = hit2df(getTestFile("hitran_2016_H2O_2iso_2000_2100cm.par"),
cache="regen")
assert len(df) > 0
# Now test nothing is loaded if asking for outside the wavenumber range
with pytest.raises(IrrelevantFileWarning):
df = hit2df(getTestFile("hitran_2016_H2O_2iso_2000_2100cm.par"),
load_wavenum_min=100000)
with pytest.raises(IrrelevantFileWarning):
df = hit2df(getTestFile("hitran_2016_H2O_2iso_2000_2100cm.par"),
load_wavenum_max=0.5)
def test_hitran_h2o(verbose=True, warnings=True, **kwargs):
# 1. Load
df = hit2df(getTestFile('hitran_2016_H2O_2iso_2000_2100cm.par'), cache='regen')
if verbose:
print('Read hitran_2016_H2O_2iso_2000_2100cm.par')
print('-----------------------------------------')
print(df.head())
# 2. Test
assert (list(df.loc[0, ['v1u', 'v2u', 'v3u', 'v1l', 'v2l', 'v3l']]) ==
[0, 2, 0, 0, 1, 0])
assert df.loc[26, 'ju'] == 5 # in .par : line 27, column 99-100
assert df.loc[27, 'ju'] == 18 # in .par : line 28, column 99-100
assert df.dtypes['v1l'] == np.int64
assert df.dtypes['v3u'] == np.int64
assert df.dtypes['ju'] == np.int64
assert df.dtypes['Kau'] == np.int64
assert df.dtypes['Kcu'] == np.int64
assert df.dtypes['jl'] == np.int64
assert df.dtypes['Kal'] == np.int64
assert df.dtypes['Kcl'] == np.int64
return True
def test_local_hitran_h2o(verbose=True, warnings=True, **kwargs):
# 1. Load
df = hit2df(getTestFile("hitran_2016_H2O_2iso_2000_2100cm.par"),
cache="regen")
if verbose:
print("Read hitran_2016_H2O_2iso_2000_2100cm.par")
print("-----------------------------------------")
print(df.head())
# 2. Test
assert list(df.loc[0, ["v1u", "v2u", "v3u", "v1l", "v2l", "v3l"]]) == [
0,
2,
0,
0,
1,
0,
]
assert df.loc[26, "ju"] == 5 # in .par : line 27, column 99-100
assert df.loc[27, "ju"] == 18 # in .par : line 28, column 99-100
assert df.dtypes["v1l"] == np.int64
assert df.dtypes["v3u"] == np.int64
assert df.dtypes["ju"] == np.int64
assert df.dtypes["Kau"] == np.int64
assert df.dtypes["Kcu"] == np.int64
assert df.dtypes["jl"] == np.int64
assert df.dtypes["Kal"] == np.int64
assert df.dtypes["Kcl"] == np.int64
return True
def test_hitran_parser(verbose=True, warnings=True, **kwargs):
''' Analyse some default files to make sure everything still works'''
t0 = time()
df = hit2df(getTestFile('hitran_CO_fragment.par'))
if verbose: print('File loaded in {0:.0f}s'.format(time()-t0))
if verbose: print(df.head())
assert (list(df.loc[0, ['v1u', 'v1l']]) == [4, 4])
t0 = time()
df = hit2df(getTestFile('hitran_CO2_fragment.par'))
if verbose: print('File loaded in {0:.0f}s'.format(time()-t0))
if verbose: print(df.head())
assert (list(df.loc[0, ['v1u', 'v2u', 'l2u', 'v3u', 'v1l', 'v2l', 'l2l', 'v3l']]) ==
[4, 0, 0, 0, 0, 0, 0, 1])
return True
def test_hitran_co2(verbose=True, warnings=True, **kwargs):
# 1. Load
df = hit2df(getTestFile('hitran_CO2_fragment.par'), cache='regen')
if verbose:
print('Read hitran_CO2_fragment.par')
print('----------------------------')
print(df.head())
# 2. Test
assert (list(df.loc[0, ['v1u', 'v2u', 'l2u', 'v3u', 'v1l', 'v2l', 'l2l', 'v3l']]) ==
[4, 0, 0, 0, 0, 0, 0, 1])
assert df.dtypes['v1l'] == np.int64
assert df.dtypes['v3u'] == np.int64
return True
def test_hitran_co(verbose=True, warnings=True, **kwargs):
''' Analyse some default files to make sure everything still works'''
# 1. Load
df = hit2df(getTestFile('hitran_CO_fragment.par'), cache='regen')
if verbose:
print('Read hitran_CO_fragment.par')
print('---------------------------')
print(df.head())
# 2. Test
assert (list(df.loc[0, ['vu', 'vl']]) == [4, 4])
assert df.dtypes['vu'] == np.int64
assert df.dtypes['vl'] == np.int64
return True
def test_local_hitran_co(verbose=True, warnings=True, **kwargs):
"""Analyse some default files to make sure everything still works."""
# 1. Load
df = hit2df(getTestFile("hitran_CO_fragment.par"), cache="regen")
if verbose:
print("Read hitran_CO_fragment.par")
print("---------------------------")
print(df.head())
# 2. Test
assert list(df.loc[0, ["vu", "vl"]]) == [4, 4]
assert df.dtypes["vu"] == np.int64
assert df.dtypes["vl"] == np.int64
return True
def test_local_hitran_co2(verbose=True, warnings=True, **kwargs):
# 1. Load
df = hit2df(getTestFile("hitran_CO2_fragment.par"), cache="regen")
if verbose:
print("Read hitran_CO2_fragment.par")
print("----------------------------")
print(df.head())
# 2. Test
assert list(
df.loc[0,
["v1u", "v2u", "l2u", "v3u", "v1l", "v2l", "l2l", "v3l"]]) == [
4, 0, 0, 0, 0, 0, 0, 1
]
assert df.dtypes["v1l"] == np.int64
assert df.dtypes["v3u"] == np.int64
return True