def test_ERAInterim_reshuffle_grb():
# test reshuffling era interim grib images to time series
inpath = os.path.join(os.path.dirname(os.path.abspath(__file__)), '..',
"ecmwf_models-test-data", "ERA-Interim", "grib")
ts_path = tempfile.mkdtemp()
startdate = '2000-01-01'
enddate = '2000-01-01'
parameters = ["swvl1"]# , "swvl2"]
h_steps = ['--h_steps', '0', '12']
args = [inpath, ts_path, startdate, enddate] + parameters + h_steps
try:
main(args)
assert len(glob.glob(os.path.join(ts_path, "*.nc"))) == 2593
ds = ERATs(ts_path, ioclass_kws={'read_bulk':True})
ts = ds.read(48, 15)
swvl1_should = np.array([0.171761, 0.171738], dtype=np.float32)
nptest.assert_allclose(ts['swvl1'].values, swvl1_should, rtol=1e-5)
#swvl2_should = np.array([0.178139, 0.178200], dtype=np.float32)
#nptest.assert_allclose(ts['swvl2'].values, swvl2_should, rtol=1e-5)
shutil.rmtree(ts_path)
except Exception as e:
shutil.rmtree(ts_path)
raise e
def test_ERAInterim_reshuffle_nc():
# test reshuffling era interim grib images to time series
inpath = os.path.join(os.path.dirname(os.path.abspath(__file__)), '..',
"ecmwf_models-test-data", "ERA-Interim", "netcdf")
startdate = '2000-01-01'
enddate = '2000-01-01'
parameters = ["swvl1", "swvl2"]
h_steps = ['--h_steps', '0', '12']
with tempfile.TemporaryDirectory() as ts_path:
args = [inpath, ts_path, startdate, enddate] + parameters + h_steps
main(args)
assert len(glob.glob(os.path.join(ts_path, "*.nc"))) == 2593
ds = ERATs(ts_path, ioclass_kws={'read_bulk': True})
ts = ds.read(48, 15)
ds.close()
swvl1_should = np.array([0.171854, 0.171738], dtype=np.float32)
nptest.assert_allclose(ts['swvl1'].values, swvl1_should, rtol=1e-5)
swvl2_should = np.array([0.179816, 0.179860], dtype=np.float32)
nptest.assert_allclose(ts['swvl2'].values, swvl2_should, rtol=1e-5)