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