def test_generate_network_taudem_dinf():
"""
Checks generate TauDEM network dinf
"""
print("TEST 13: TEST GENERATE TauDEM NETWORK DINF")
TAUDEM_EXE_PATH = os.path.join(MAIN_TESTS_FOLDER,
"..", "..", "TauDEM")
td = TauDEM(TAUDEM_EXE_PATH)
elevation_dem = os.path.join(GIS_INPUT_DATA_PATH, 'jamaica_dem.tif')
td.demToStreamNetwork(OUTPUT_DATA_PATH,
pit_filled_elevation_grid=elevation_dem,
threshold=1000,
use_dinf=True)
ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'flow_dir_grid_d8.tif')))
ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'flow_dir_grid_d8.prj')))
ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'flow_dir_grid_dinf.tif')))
ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'flow_dir_grid_dinf.prj')))
ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'contributing_area_grid_d8.tif')))
ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'contributing_area_grid_d8.prj')))
ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'contributing_area_grid_dinf.tif')))
ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'contributing_area_grid_dinf.prj')))
ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'slope_grid_d8.tif')))
ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'slope_grid_d8.prj')))
ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'slope_grid_dinf.tif')))
ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'slope_grid_dinf.prj')))
ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'stream_raster_grid.tif')))
ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'stream_raster_grid.prj')))
ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'stream_order_grid.tif')))
ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'stream_order_grid.prj')))
ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'network_connectivity_tree.txt')))
ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'network_coordinates.txt')))
# ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'stream_reach_file.shp')))
# ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'stream_reach_file.shx')))
# ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'stream_reach_file.dbf')))
ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'stream_reach_file.prj')))
ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'watershed_grid.tif')))
ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'watershed_grid.prj')))
ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'watershed_shapefile.shp')))
ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'watershed_shapefile.shx')))
ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'watershed_shapefile.dbf')))
ok_(os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'watershed_shapefile.prj')))
#cleanup
remove_files(*[f for f in glob(os.path.join(OUTPUT_DATA_PATH,"*")) if not f.endswith(".gitignore")])
def test_generate_network_taudem_dinf():
"""
Checks generate TauDEM network dinf
"""
print("TEST 13: TEST GENERATE TauDEM NETWORK DINF")
td = TauDEM(TAUDEM_EXE_PATH)
elevation_dem = os.path.join(GIS_INPUT_DATA_PATH, 'jamaica_dem.tif')
td.demToStreamNetwork(OUTPUT_DATA_PATH,
pit_filled_elevation_grid=elevation_dem,
threshold=1000,
use_dinf=True)
assert (os.path.exists(
os.path.join(OUTPUT_DATA_PATH, 'flow_dir_grid_d8.tif')))
assert (os.path.exists(
os.path.join(OUTPUT_DATA_PATH, 'flow_dir_grid_d8.prj')))
assert (os.path.exists(
os.path.join(OUTPUT_DATA_PATH, 'flow_dir_grid_dinf.tif')))
assert (os.path.exists(
os.path.join(OUTPUT_DATA_PATH, 'flow_dir_grid_dinf.prj')))
assert (os.path.exists(
os.path.join(OUTPUT_DATA_PATH, 'contributing_area_grid_d8.tif')))
assert (os.path.exists(
os.path.join(OUTPUT_DATA_PATH, 'contributing_area_grid_d8.prj')))
assert (os.path.exists(
os.path.join(OUTPUT_DATA_PATH, 'contributing_area_grid_dinf.tif')))
assert (os.path.exists(
os.path.join(OUTPUT_DATA_PATH, 'contributing_area_grid_dinf.prj')))
assert (os.path.exists(os.path.join(OUTPUT_DATA_PATH,
'slope_grid_d8.tif')))
assert (os.path.exists(os.path.join(OUTPUT_DATA_PATH,
'slope_grid_d8.prj')))
assert (os.path.exists(
os.path.join(OUTPUT_DATA_PATH, 'slope_grid_dinf.tif')))
assert (os.path.exists(
os.path.join(OUTPUT_DATA_PATH, 'slope_grid_dinf.prj')))
assert (os.path.exists(
os.path.join(OUTPUT_DATA_PATH, 'stream_raster_grid.tif')))
assert (os.path.exists(
os.path.join(OUTPUT_DATA_PATH, 'stream_raster_grid.prj')))
assert (os.path.exists(
os.path.join(OUTPUT_DATA_PATH, 'stream_order_grid.tif')))
assert (os.path.exists(
os.path.join(OUTPUT_DATA_PATH, 'stream_order_grid.prj')))
assert (os.path.exists(
os.path.join(OUTPUT_DATA_PATH, 'network_connectivity_tree.txt')))
assert (os.path.exists(
os.path.join(OUTPUT_DATA_PATH, 'network_coordinates.txt')))
# assert (os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'stream_reach_file.shp')))
# assert (os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'stream_reach_file.shx')))
# assert (os.path.exists(os.path.join(OUTPUT_DATA_PATH, 'stream_reach_file.dbf')))
assert (os.path.exists(
os.path.join(OUTPUT_DATA_PATH, 'stream_reach_file.prj')))
assert (os.path.exists(os.path.join(OUTPUT_DATA_PATH,
'watershed_grid.tif')))
assert (os.path.exists(os.path.join(OUTPUT_DATA_PATH,
'watershed_grid.prj')))
assert (os.path.exists(
os.path.join(OUTPUT_DATA_PATH, 'watershed_shapefile.shp')))
assert (os.path.exists(
os.path.join(OUTPUT_DATA_PATH, 'watershed_shapefile.shx')))
assert (os.path.exists(
os.path.join(OUTPUT_DATA_PATH, 'watershed_shapefile.dbf')))
assert (os.path.exists(
os.path.join(OUTPUT_DATA_PATH, 'watershed_shapefile.prj')))
#cleanup
remove_files(*[
f for f in glob(os.path.join(OUTPUT_DATA_PATH, "*"))
if not f.endswith(".gitignore")
])
