def test_landfire_200(self, mock_reproject):
config = self.base_config_copy()
config.raw_cfg['generate_topo'][
'vegetation_dataset'] = 'landfire_2.0.0' # noqa
config.raw_cfg['generate_topo'][
'vegetation_folder'] = '../../landfire/landfire_200' # noqa
config.apply_recipes()
config = cast_all_variables(config, config.mcfg)
gt = GenerateTopo(config_file=config)
gt.run()
self.assertTrue(mock_reproject.called)
self.assertTrue(mock_reproject.call_count == 1)
ds = xr.open_dataset(os.path.join(self.basin_dir, 'output', 'topo.nc'))
self.assertCountEqual(list(ds.coords.keys()), ['y', 'x'])
self.assertCountEqual(list(ds.keys()), [
'dem', 'mask', 'veg_height', 'veg_k', 'veg_tau', 'veg_type',
'projection'
])
self.compare_netcdf_files('landfire_200/topo.nc', 'topo.nc')
def test_run(self, mock_veg):
gt = GenerateTopo(config_file=self.config_file)
gt.run()
ds = xr.open_dataset(os.path.join(self.basin_dir, 'output', 'topo.nc'))
self.assertCountEqual(list(ds.coords.keys()), ['y', 'x'])
self.assertCountEqual(list(ds.keys()), [
'dem', 'mask', 'veg_height', 'veg_k', 'veg_tau', 'veg_type',
'projection'
])
self.compare_netcdf_files('landfire_140/topo.nc', 'topo.nc')
def main():
# Parge command line arguments
p = argparse.ArgumentParser(
description='Setup a new basin for SMRF/AWSM.'
' Creates all the required static files'
' required for running a basin. The output is a'
' topo.nc containing: dem, mask, veg height,'
' veg type, veg tau, and veg k')
p.add_argument(dest='config_file', help="Path to configuration file")
args = p.parse_args()
gt = GenerateTopo(args.config_file)
gt.run()
def test_no_veg(self, mock_reproject):
gt = GenerateTopo(config_file=self.config_file)
gt.config['vegetation_dataset'] = None
gt.run()
self.assertFalse(mock_reproject.called)
self.assertTrue(mock_reproject.call_count == 0)
ds = xr.open_dataset(os.path.join(self.basin_dir, 'output', 'topo.nc'))
self.assertCountEqual(list(ds.coords.keys()), ['y', 'x'])
self.assertCountEqual(list(ds.keys()), [
'dem', 'mask', 'veg_height', 'veg_k', 'veg_tau', 'veg_type',
'projection'
])
for image in gt.veg.VEG_IMAGES:
if image == 'veg_type':
# all veg type are 0 values
self.assertTrue(np.sum(ds[image].isnull().values) == 0)
else:
self.assertTrue(np.all(ds[image].isnull().values))
def setUp(self) -> None:
self.subject = GenerateTopo(config_file=self.config_file)
class TestBasinSetup(BasinSetupLakes):
# TODO change extents to this instead of decimals
# EXTENTS = [318520.0, 4157550.0, 329470.0, 4167900.0]
EXTENTS = [319570.405027, 4157787.07547, 328270.405027, 4167087.07547]
EXTENTS_RASTER = [319570.405, 4157787.075, 328270.405, 4167087.075]
@classmethod
def setUpClass(self):
super().setUpClass()
def setUp(self) -> None:
self.subject = GenerateTopo(config_file=self.config_file)
def test_init(self):
self.assertIsInstance(self.subject.ucfg, UserConfig)
def test_set_extents(self):
self.subject.set_extents()
self.assertListEqual(self.subject.extents, self.EXTENTS)
self.assertIsInstance(self.subject.transform, Affine)
self.assertTrue(len(self.subject.x) == 58)
self.assertTrue(len(self.subject.y) == 62)
def test_load_basin_shapefiles(self):
self.subject.load_basin_shapefiles()
self.assertTrue(len(self.subject.basin_shapefiles) == 1)
self.assertIsInstance(self.subject.basin_shapefiles[0], Shapefile)
self.assertDictEqual(self.subject.basin_shapefiles[0].crs, self.CRS)
def test_load_dem(self):
self.subject.crs = self.CRS
self.subject.extents = self.EXTENTS
self.subject.load_dem()
extents, cell_size = domain_extent.parse_from_file(
'tests/Lakes/output/temp/clipped_dem.tif')
self.assertListEqual(extents, self.EXTENTS_RASTER)
self.assertTrue(cell_size == self.subject.config['cell_size'])
self.assertIsInstance(self.subject.dem, xr.DataArray)
self.assertCountEqual(list(self.subject.dem.coords.keys()),
['y', 'x', 'spatial_ref'])
@patch.object(Landfire140, 'reproject', return_value=True)
def test_load_vegetation(self, mock_veg):
self.subject.crs = self.CRS
self.subject.extents = self.EXTENTS
self.subject.load_vegetation()
extents, cell_size = domain_extent.parse_from_file(
self.subject.veg.clipped_images['veg_type'])
self.assertListEqual(extents, self.EXTENTS_RASTER)
self.assertTrue(cell_size == self.subject.config['cell_size'])
self.assertIsInstance(self.subject.veg, Landfire140)
self.assertCountEqual(list(self.subject.veg.veg_tau_k.coords.keys()),
['y', 'x', 'spatial_ref'])
self.assertCountEqual(list(self.subject.veg.veg_height.coords.keys()),
['y', 'x', 'spatial_ref'])
@patch.object(Landfire140, 'reproject', return_value=True)
def test_run(self, mock_veg):
gt = GenerateTopo(config_file=self.config_file)
gt.run()
ds = xr.open_dataset(os.path.join(self.basin_dir, 'output', 'topo.nc'))
self.assertCountEqual(list(ds.coords.keys()), ['y', 'x'])
self.assertCountEqual(list(ds.keys()), [
'dem', 'mask', 'veg_height', 'veg_k', 'veg_tau', 'veg_type',
'projection'
])
self.compare_netcdf_files('landfire_140/topo.nc', 'topo.nc')