def test_basin_filename(): myEnv = modelEnv.ModelEnv(topDir=topDir) assert_equals( myEnv.fixed_filename(type='basin_mask', tileID='h23v05', drainageID='IN_Indus_at_Kotri'), topDir + '/basin_masks/IN_Indus_at_Kotri.basin_mask.h23v05.tif')
def test_scag_filename(): myEnv = modelEnv.ModelEnv(topDir=topDir) assert_equals( myEnv.forcing_filename(type='modscag_gf', tileID='h23v05', yyyy=2005), topDir + '/MODSCAG_GF/v04/h23v05/MODSCAG_GF_Snow.v0.4.h23v05_2005.h5')
def test_scag_filename_no_yyyy(): myEnv = modelEnv.ModelEnv(topDir=topDir) assert_raises(ValueError, myEnv.forcing_filename, type='modscag_gf', tileID='h23v05')
def test_modice_filename(): myEnv = modelEnv.ModelEnv(topDir=topDir) assert_equals( myEnv.fixed_filename(type='modice', tileID='h23v05'), topDir + '/modicev04/MODICE.v0.4.h23v05.3strike.min05yr.mask.nc')
def test_dem_filename(): myEnv = modelEnv.ModelEnv(topDir=topDir) assert_equals(myEnv.fixed_filename(type='dem', tileID='h23v05'), topDir + '/SRTMGL3/SRTMGL3.v0.1.h23v05.tif')
def test_basin_filename_no_drainageID(): myEnv = modelEnv.ModelEnv(topDir=topDir) assert_raises(ValueError, myEnv.fixed_filename, type='basin_mask')
def test_modice_filename_no_tileID(): myEnv = modelEnv.ModelEnv(topDir=topDir) assert_raises(ValueError, myEnv.fixed_filename)
def test_set_model_top_dir(): myEnv = modelEnv.ModelEnv() assert_equals(myEnv.tileConfig['model_top_dir'], "/projects/CHARIS") assert_equals(myEnv.set_model_top_dir(topDir=topDir), True) assert_equals(myEnv.tileConfig['model_top_dir'], topDir)
def test_init_modelEnv_with_topDir(): myEnv = modelEnv.ModelEnv(topDir='/test') assert_equals(myEnv.tileConfig['model_top_dir'], "/test")
# coding: utf-8 # In[1]: from imp import reload import modelEnv reload(modelEnv) myEnv = modelEnv.ModelEnv( topDir='/Users/brodzik/projects/CHARIS/charis_training_2015_data', verbose=True) print myEnv.fixed_filename(type='modice', tileID='h23v05', verbose=True) # In[2]: from netCDF4 import Dataset modice_filename = myEnv.fixed_filename(type='modice', tileID='h23v05', verbose=True) fid = Dataset(modice_filename, 'r', format='NETCDF4') fid # In[3]: modice = fid.variables['modice_min_year_mask'] modice # In[4]: data = modice[:] data.shape