def test_weight_table_with_area_id():
"""
Checks generating weight table with area id
"""
print("TEST 17: TEST GENERATE WEIGHT TABLE WITH INVALID POLYGON")
generated_weight_table_file = os.path.join(OUTPUT_DATA_PATH,
"weight_area.csv")
# rapid_connect
rapid_connect_file = os.path.join(COMPARE_DATA_PATH, "x-x",
"rapid_connect.csv")
lsm_grid = os.path.join(LSM_INPUT_DATA_PATH, "gldas2",
"GLDAS_NOAH025_3H.A20101231.0000.020.nc4")
CreateWeightTableLDAS(in_ldas_nc=lsm_grid,
in_nc_lon_var="lon",
in_nc_lat_var="lat",
in_catchment_shapefile=os.path.join(
GIS_INPUT_DATA_PATH, 'test_catchments.shp'),
river_id="DrainLnID",
area_id="Shape_Area",
in_connectivity_file=rapid_connect_file,
out_weight_table=generated_weight_table_file)
generated_weight_table_file_solution = os.path.join(
COMPARE_DATA_PATH, "x-x", "weight_area.csv")
assert (compare_csv_decimal_files(generated_weight_table_file,
generated_weight_table_file_solution))
remove_files(generated_weight_table_file)
def test_gen_weight_table_joules():
"""
Checks generating weight table for Joules grid
"""
print("TEST 9: TEST GENERATE WEIGHT TABLE FOR Joules GRIDS")
generated_weight_table_file = os.path.join(OUTPUT_DATA_PATH,
"weight_joules.csv")
#rapid_connect
rapid_connect_file = os.path.join(COMPARE_DATA_PATH, "u-k",
"rapid_connect.csv")
lsm_grid = os.path.join(LSM_INPUT_DATA_PATH, "joules",
"ukv_test.runoff.20080803_00.nc")
CreateWeightTableLDAS(in_ldas_nc=lsm_grid,
in_nc_lon_var="east_west",
in_nc_lat_var="north_south",
in_catchment_shapefile=os.path.join(
GIS_INPUT_DATA_PATH, 'u-k',
'CatchmentSubset.shp'),
river_id="DrainLnID",
in_connectivity_file=rapid_connect_file,
out_weight_table=generated_weight_table_file)
generated_weight_table_file_solution = os.path.join(
COMPARE_DATA_PATH, "u-k", "weight_joules.csv")
assert (compare_csv_decimal_files(generated_weight_table_file,
generated_weight_table_file_solution))
remove_files(generated_weight_table_file)
def test_gen_weight_table_lis_no_intersect():
"""
Checks generating weight table for LIS grid with no intersect
"""
print("TEST 8: TEST GENERATE WEIGHT TABLE FOR LIS GRIDS WITH NO INTERSECT")
generated_weight_table_file = os.path.join(OUTPUT_DATA_PATH,
"weight_lis_no_intersect.csv")
#rapid_connect
rapid_connect_file = os.path.join(GIS_INPUT_DATA_PATH, "uk-no_intersect",
"rapid_connect_45390.csv")
lsm_grid = os.path.join(LSM_INPUT_DATA_PATH, "lis",
"LIS_HIST_201101210000.d01.nc")
CreateWeightTableLDAS(in_ldas_nc=lsm_grid,
in_nc_lon_var="lon",
in_nc_lat_var="lat",
in_catchment_shapefile=os.path.join(
GIS_INPUT_DATA_PATH, 'uk-no_intersect',
'Catchment_thames_drainID45390.shp'),
river_id="DrainLnID",
in_connectivity_file=rapid_connect_file,
out_weight_table=generated_weight_table_file)
generated_weight_table_file_solution = os.path.join(
COMPARE_DATA_PATH, "uk-no_intersect", "weight_lis_no_intersect.csv")
assert (compare_csv_decimal_files(generated_weight_table_file,
generated_weight_table_file_solution))
remove_files(generated_weight_table_file)
def test_gen_weight_table_gldas2():
"""
Checks generating weight table for GLDAS V2 grid
"""
print("TEST 6: TEST GENERATE WEIGHT TABLE FOR GLDAS V2 GRIDS")
generated_weight_table_file = os.path.join(OUTPUT_DATA_PATH,
"weight_gldas2.csv")
#rapid_connect
rapid_connect_file = os.path.join(COMPARE_DATA_PATH, "x-x",
"rapid_connect.csv")
lsm_grid = os.path.join(LSM_INPUT_DATA_PATH, "gldas2",
"GLDAS_NOAH025_3H.A20101231.0000.020.nc4")
CreateWeightTableLDAS(in_ldas_nc=lsm_grid,
in_nc_lon_var="lon",
in_nc_lat_var="lat",
in_catchment_shapefile=os.path.join(
GIS_INPUT_DATA_PATH, 'catchment.shp'),
river_id="FEATUREID",
in_connectivity_file=rapid_connect_file,
out_weight_table=generated_weight_table_file)
generated_weight_table_file_solution = os.path.join(
COMPARE_DATA_PATH, "x-x", "weight_gldas2.csv")
assert (compare_csv_decimal_files(generated_weight_table_file,
generated_weight_table_file_solution))
remove_files(generated_weight_table_file)
def test_gen_weight_table_lis():
"""
Checks generating weight table for LIS grid
"""
print("TEST 7: TEST GENERATE WEIGTH TABLE FOR LIS GRIDS")
generated_weight_table_file = os.path.join(OUTPUT_DATA_PATH,
"weight_lis.csv")
#rapid_connect
rapid_connect_file = os.path.join(COMPARE_DATA_PATH, "u-k",
"rapid_connect.csv")
lsm_grid = os.path.join(LSM_INPUT_DATA_PATH, "lis", "LIS_HIST_201101210000.d01.nc")
CreateWeightTableLDAS(in_ldas_nc=lsm_grid,
in_nc_lon_var="lon",
in_nc_lat_var="lat",
in_catchment_shapefile=os.path.join(GIS_INPUT_DATA_PATH, 'u-k', 'CatchmentSubset.shp'),
river_id="DrainLnID",
in_connectivity_file=rapid_connect_file,
out_weight_table=generated_weight_table_file)
generated_weight_table_file_solution = os.path.join(COMPARE_DATA_PATH, "u-k",
"weight_lis.csv")
ok_(compare_csv_decimal_files(generated_weight_table_file,
generated_weight_table_file_solution))
remove_files(generated_weight_table_file)
def test_gen_weight_table_lis_land_mask():
"""
Checks generating weight table for LIS grid with land mask.
"""
print("TEST 20: TEST GENERATE WEIGHT TABLE FOR LIS GRID WITH LAND MASK.")
generated_weight_table_file = os.path.join(
OUTPUT_DATA_PATH, "weight_lis_land_fraction_mendocino_subset.csv")
#rapid_connect
rapid_connect_file = os.path.join(
COMPARE_DATA_PATH, "mendocino_nhdplus_catchment",
"rapid_connectivity_mendocino_sample.csv")
lsm_grid = os.path.join(LSM_INPUT_DATA_PATH, "lis_land_mask",
"lisglobalmask557ww_mendocino_subset.nc")
CreateWeightTableLDAS(in_ldas_nc=lsm_grid,
in_nc_lon_var='lon',
in_nc_lat_var='lat',
in_catchment_shapefile=os.path.join(
GIS_INPUT_DATA_PATH,
'mendocino_nhdplus_catchment',
'NHDCat_mendocino_watershed_hopland_sample.shp'),
river_id='FEATUREID',
in_connectivity_file=rapid_connect_file,
out_weight_table=generated_weight_table_file,
in_land_area_fraction_var='LANDMASK')
generated_weight_table_file_solution = os.path.join(
COMPARE_DATA_PATH, 'mendocino_nhdplus_catchment',
'weight_lis_land_fraction_mendocino_subset.csv')
assert (compare_csv_decimal_files(generated_weight_table_file,
generated_weight_table_file_solution))
remove_files(generated_weight_table_file)