def test_refet_hourly_func_output(hourly_params):
"""Test hourly RefET calculation at a single point and time"""
# If I don't copy, the pop changes the test values in hourly_data()
inputs = hourly_params.copy()
surface = inputs.pop('surface')
expected = inputs.pop('expected')
print('ETr: {}'.format(expected))
# Cast all numeric inputs to ee.Number type except tmean (for now)
adj_inputs = {}
for k, v in inputs.items():
if k == 'rso_type':
adj_inputs[k] = v
elif k == 'tmean':
adj_inputs[k] = ee.Image.constant(v)
else:
adj_inputs[k] = ee.Number(v)
if surface.lower() == 'etr':
refet = Hourly(**adj_inputs).etr
elif surface.lower() == 'eto':
refet = Hourly(**adj_inputs).eto
output = refet\
.reduceRegion(ee.Reducer.first(), geometry=constant_geom, scale=1)\
.getInfo()
assert float(output[surface.lower()]) == pytest.approx(expected, abs=0.01)
def test_refet_daily_etsz(surface, expected):
refet = Hourly(
tmean=ee.Image.constant(h_args['tmean']),
ea=ee.Image.constant(h_args['ea']),
rs=ee.Image.constant(h_args['rs']), uz=ee.Image.constant(h_args['uz']),
zw=ee.Image.constant(s_args['zw']), elev=ee.Number(s_args['elev']),
lat=ee.Number(s_args['lat']), lon=ee.Number(s_args['lon']),
doy=ee.Number(h_args['doy']), time=ee.Number(h_args['time']),
method='refet')
output = refet.etsz(surface).rename(['etsz']).reduceRegion(
reducer=ee.Reducer.first(),
geometry=ee.Geometry.Rectangle([0, 0, 10, 10], 'EPSG:32613', False),
scale=1).getInfo()
assert float(output['etsz']) == pytest.approx(expected)
def test_refet_hourly_input_positions():
refet = Hourly(
ee.Image.constant(h_args['tmean']), ee.Image.constant(h_args['ea']),
ee.Image.constant(h_args['rs']), ee.Image.constant(h_args['uz']),
ee.Image.constant(s_args['zw']), ee.Number(s_args['elev']),
ee.Number(s_args['lat']), ee.Number(s_args['lon']),
ee.Number(h_args['doy']), ee.Number(h_args['time']), method='refet')
output = refet.etr\
.reduceRegion(ee.Reducer.first(), geometry=constant_geom, scale=1)\
.getInfo()
assert float(output['etr']) == pytest.approx(h_args['etr_refet'])
def test_refet_hourly_default_method_etr():
refet = Hourly(
tmean=ee.Image.constant(h_args['tmean']),
ea=ee.Image.constant(h_args['ea']),
rs=ee.Image.constant(h_args['rs']), uz=ee.Image.constant(h_args['uz']),
zw=ee.Image.constant(s_args['zw']), elev=ee.Number(s_args['elev']),
lat=ee.Number(s_args['lat']), lon=ee.Number(s_args['lon']),
doy=ee.Number(h_args['doy']), time=ee.Number(h_args['time']))
output = refet.etr\
.reduceRegion(ee.Reducer.first(), geometry=constant_geom, scale=1)\
.getInfo()
assert float(output['etr']) == pytest.approx(h_args['etr_asce'])
def test_refet_hourly_rtma_etr():
"""Generate a fake RTMA image from the test values"""
rtma_time = ee.Date(
'2015-07-01T{}:00:00'.format(int(h_args['time'])), 'UTC').millis()
rtma_img = ee.Image.constant([
h_args['tmean'], h_args['q_asce'], h_args['uz']])\
.rename(['TMP', 'SPFH', 'WIND'])\
.set('system:time_start', rtma_time)
refet = Hourly.rtma(
ee.Image(rtma_img),
rs=ee.Image.constant(h_args['rs']),
elev=ee.Number(s_args['elev']),
lat=ee.Number(s_args['lat']), lon=ee.Number(s_args['lon']),
zw=ee.Number(s_args['zw']), method='asce')
output = refet.etr\
.reduceRegion(ee.Reducer.first(), geometry=constant_geom, scale=1)\
.getInfo()
assert float(output['etr']) == pytest.approx(h_args['etr_asce'])
def test_refet_hourly_nldas_etr():
"""Generate a fake NLDAS image from the test values"""
nldas_time = ee.Date(
'2015-07-01T{}:00:00'.format(int(h_args['time'])), 'UTC').millis()
wind_u = h_args['uz'] / (2 ** 0.5)
nldas_img = ee.Image.constant([
h_args['tmean'], h_args['q_asce'], h_args['rs'] / 0.0036,
wind_u, wind_u])\
.rename(['temperature', 'specific_humidity', 'shortwave_radiation',
'wind_u', 'wind_v'])\
.set('system:time_start', nldas_time)
refet = Hourly.nldas(
ee.Image(nldas_img), elev=ee.Number(s_args['elev']),
lat=ee.Number(s_args['lat']), lon=ee.Number(s_args['lon']),
zw=ee.Number(s_args['zw']), method='asce')
output = refet.etr\
.reduceRegion(ee.Reducer.first(), geometry=constant_geom, scale=1)\
.getInfo()
assert float(output['etr']) == pytest.approx(h_args['etr_asce'])