def test_biophysical_aoi(self):
"""WaveEnergy: testing Biophysical component with an AOI."""
from natcap.invest.wave_energy import wave_energy
args = WaveEnergyRegressionTests.generate_base_args(self.workspace_dir)
args['aoi_uri'] = os.path.join(SAMPLE_DATA, 'WaveEnergy', 'input',
'AOI_WCVI.shp')
wave_energy.execute(args)
raster_results = [
'wp_rc.tif', 'wp_kw.tif', 'capwe_rc.tif', 'capwe_mwh.tif'
]
for raster_path in raster_results:
natcap.invest.pygeoprocessing_0_3_3.testing.assert_rasters_equal(
os.path.join(args['workspace_dir'], 'output', raster_path),
os.path.join(REGRESSION_DATA, 'aoi', raster_path), 1e-9)
table_results = ['capwe_rc.csv', 'wp_rc.csv']
for table_path in table_results:
natcap.invest.pygeoprocessing_0_3_3.testing.assert_csv_equal(
os.path.join(args['workspace_dir'], 'output', table_path),
os.path.join(REGRESSION_DATA, 'aoi', table_path), 1e-9)
def test_removing_filenames(self):
"""WaveEnergy: testing file paths which already exist are removed."""
from natcap.invest.wave_energy import wave_energy
workspace_dir = 'test_removing_filenames'
args = WaveEnergyRegressionTests.generate_base_args(
workspace_dir) #self.workspace_dir)
args['aoi_uri'] = os.path.join(SAMPLE_DATA, 'WaveEnergy', 'input',
'AOI_WCVI.shp')
args['valuation_container'] = True
args['land_gridPts_uri'] = os.path.join(SAMPLE_DATA, 'WaveEnergy',
'input',
'LandGridPts_WCVI.csv')
args['machine_econ_uri'] = os.path.join(
SAMPLE_DATA, 'WaveEnergy', 'input', 'Machine_Pelamis_Economic.csv')
args['number_of_machines'] = 28
wave_energy.execute(args)
# Run through the model again, which should mean deleting
# shapefiles that have already been made, but which need
# to be created again.
wave_energy.execute(args)
raster_results = [
'wp_rc.tif', 'wp_kw.tif', 'capwe_rc.tif', 'capwe_mwh.tif',
'npv_rc.tif', 'npv_usd.tif'
]
for raster_path in raster_results:
natcap.invest.pygeoprocessing_0_3_3.testing.assert_rasters_equal(
os.path.join(args['workspace_dir'], 'output', raster_path),
os.path.join(REGRESSION_DATA, 'valuation', raster_path), 1e-9)
vector_results = ['GridPts_prj.shp', 'LandPts_prj.shp']
for vector_path in vector_results:
natcap.invest.pygeoprocessing_0_3_3.testing.assert_vectors_equal(
os.path.join(args['workspace_dir'], 'output', vector_path),
os.path.join(REGRESSION_DATA, 'valuation', vector_path))
table_results = ['capwe_rc.csv', 'wp_rc.csv', 'npv_rc.csv']
for table_path in table_results:
natcap.invest.pygeoprocessing_0_3_3.testing.assert_csv_equal(
os.path.join(args['workspace_dir'], 'output', table_path),
os.path.join(REGRESSION_DATA, 'valuation', table_path))
def test_valuation_suffix_underscore(self):
"""WaveEnergy: testing suffix with an underscore through Valuation."""
from natcap.invest.wave_energy import wave_energy
args = WaveEnergyRegressionTests.generate_base_args(self.workspace_dir)
args['aoi_uri'] = os.path.join(SAMPLE_DATA, 'WaveEnergy', 'input',
'AOI_WCVI.shp')
args['valuation_container'] = True
args['land_gridPts_uri'] = os.path.join(SAMPLE_DATA, 'WaveEnergy',
'input',
'LandGridPts_WCVI.csv')
args['machine_econ_uri'] = os.path.join(
SAMPLE_DATA, 'WaveEnergy', 'input', 'Machine_Pelamis_Economic.csv')
args['number_of_machines'] = 28
args['suffix'] = '_val'
wave_energy.execute(args)
raster_results = [
'wp_rc_val.tif', 'wp_kw_val.tif', 'capwe_rc_val.tif',
'capwe_mwh_val.tif', 'npv_rc_val.tif', 'npv_usd_val.tif'
]
for raster_path in raster_results:
self.assertTrue(
os.path.exists(
os.path.join(args['workspace_dir'], 'output',
raster_path)))
vector_results = ['GridPts_prj_val.shp', 'LandPts_prj_val.shp']
for vector_path in vector_results:
self.assertTrue(
os.path.exists(
os.path.join(args['workspace_dir'], 'output',
vector_path)))
table_results = ['capwe_rc_val.csv', 'wp_rc_val.csv', 'npv_rc_val.csv']
for table_path in table_results:
self.assertTrue(
os.path.exists(
os.path.join(args['workspace_dir'], 'output', table_path)))
def test_valuation(self):
"""WaveEnergy: testing valuation component."""
from natcap.invest.wave_energy import wave_energy
args = WaveEnergyRegressionTests.generate_base_args(self.workspace_dir)
args['aoi_uri'] = os.path.join(SAMPLE_DATA, 'WaveEnergy', 'input',
'AOI_WCVI.shp')
args['valuation_container'] = True
args['land_gridPts_uri'] = os.path.join(SAMPLE_DATA, 'WaveEnergy',
'input',
'LandGridPts_WCVI.csv')
args['machine_econ_uri'] = os.path.join(
SAMPLE_DATA, 'WaveEnergy', 'input', 'Machine_Pelamis_Economic.csv')
args['number_of_machines'] = 28
wave_energy.execute(args)
raster_results = [
'wp_rc.tif', 'wp_kw.tif', 'capwe_rc.tif', 'capwe_mwh.tif',
'npv_rc.tif', 'npv_usd.tif'
]
for raster_path in raster_results:
natcap.invest.pygeoprocessing_0_3_3.testing.assert_rasters_equal(
os.path.join(args['workspace_dir'], 'output', raster_path),
os.path.join(REGRESSION_DATA, 'valuation', raster_path), 1e-9)
vector_results = ['GridPts_prj.shp', 'LandPts_prj.shp']
for vector_path in vector_results:
natcap.invest.pygeoprocessing_0_3_3.testing.assert_vectors_equal(
os.path.join(args['workspace_dir'], 'output', vector_path),
os.path.join(REGRESSION_DATA, 'valuation', vector_path))
table_results = ['capwe_rc.csv', 'wp_rc.csv', 'npv_rc.csv']
for table_path in table_results:
natcap.invest.pygeoprocessing_0_3_3.testing.assert_csv_equal(
os.path.join(args['workspace_dir'], 'output', table_path),
os.path.join(REGRESSION_DATA, 'valuation', table_path))