def test_checkTerm():
term1 = 'precipMONTH'
layers = LM.validateLayers(cmodel)
term, tterm, timeField = LM.checkTerm(term1, layers)
assert(term == "self.layerdict['precip'].getData()" and tterm == '' and timeField == 'MONTH'), 'checkTerm did not work properly for precipMonth'
term2 = 'MWextrajunk'
term, tterm, timeField = LM.checkTerm(term2, layers)
assert(term == "self.shakemap.getEventDict()['magnitude']extrajunk" and tterm == 'extrajunk'), 'checkTerm did not identify extrajunk in MWextrajunk term'
def test_validateLayers():
data = {'cti1': os.path.join(datadir, 'test_cti1.bil'),
'friction': os.path.join(datadir, 'test_friction.bil'),
'precip': [os.path.join(datadir, 'test_precip/prec_Jan.bil')],
'slope': os.path.join(datadir, 'test_slope.bil'),
'vs30': os.path.join(datadir, 'test_vs30.bil')}
layers = LM.validateLayers(cmodel)
assert(data == layers), 'validateLayers did not return expected result'
def test_validateUnits():
data = {'cti1': 'unitless',
'friction': 'degrees',
'precip': 'mm',
'slope': 'degrees',
'vs30': 'm/s'}
layers = LM.validateLayers(cmodel)
interp = LM.validateUnits(cmodel, layers)
assert(data == interp), 'validateInterpolations did not return expected dictionary'
def test_validateInterpolations():
data = {'cti1': 'nearest',
'friction': 'nearest',
'precip': 'nearest',
'slope': 'nearest',
'vs30': 'nearest'}
layers = LM.validateLayers(cmodel)
interp = LM.validateInterpolations(cmodel, layers)
assert(data == interp), 'validateInterpolations did not return expected dictionary'
def test_validateTerms():
data = {'b1': "self.layerdict['friction'].getData()", 'b2': "self.layerdict['slope'].getData()/100.",
'b3': "np.log(self.layerdict['vs30'].getData())", 'b4': "self.layerdict['cti1'].getData()",
'b5': "self.layerdict['precip'].getData()"}
timeField = 'MONTH'
coeff = LM.validateCoefficients(cmodel)
layers = LM.validateLayers(cmodel)
terms, time = LM.validateTerms(cmodel, coeff, layers)
assert(time == timeField), 'validateTerms did not return expected timeField'
assert(data == terms), 'validateTerms did not return expected terms'
def test_validateLayers():
data = {
'cti1': os.path.join(datadir, 'test_cti1.bil'),
'friction': os.path.join(datadir, 'test_friction.bil'),
'precip': [os.path.join(datadir, 'test_precip/prec_Jan.bil')],
'slope': os.path.join(datadir, 'test_slope.bil'),
'vs30': os.path.join(datadir, 'test_vs30.bil')
}
layers = LM.validateLayers(cmodel)
assert (data == layers), 'validateLayers did not return expected result'
def test_checkTerm():
term1 = 'precipMONTH'
layers = LM.validateLayers(cmodel)
term, tterm, timeField = LM.checkTerm(term1, layers)
assert (term == "self.layerdict['precip'].getData()" and tterm == ''
and timeField
== 'MONTH'), 'checkTerm did not work properly for precipMonth'
term2 = 'MWextrajunk'
term, tterm, timeField = LM.checkTerm(term2, layers)
assert (term == "self.shakemap.getEventDict()['magnitude']extrajunk"
and tterm == 'extrajunk'
), 'checkTerm did not identify extrajunk in MWextrajunk term'
def test_validateUnits():
data = {
'cti1': 'unitless',
'friction': 'degrees',
'precip': 'mm',
'slope': 'degrees',
'vs30': 'm/s'
}
layers = LM.validateLayers(cmodel)
interp = LM.validateUnits(cmodel, layers)
assert (data == interp
), 'validateInterpolations did not return expected dictionary'
def test_validateInterpolations():
data = {
'cti1': 'nearest',
'friction': 'nearest',
'precip': 'nearest',
'slope': 'nearest',
'vs30': 'nearest'
}
layers = LM.validateLayers(cmodel)
interp = LM.validateInterpolations(cmodel, layers)
assert (data == interp
), 'validateInterpolations did not return expected dictionary'
def test_validateTerms():
data = {
'b1': "np.nan_to_num(self.layerdict['friction'].getData())",
'b2': "self.layerdict['slope'].getData()/100.",
'b3': "np.log(self.layerdict['vs30'].getData())",
'b4': "self.layerdict['cti1'].getData()",
'b5': "self.layerdict['precip'].getData()"
}
timeField = 'MONTH'
coeff = LM.validateCoefficients(cmodel)
layers = LM.validateLayers(cmodel)
terms, time = LM.validateTerms(cmodel, coeff, layers)
assert (
time == timeField), 'validateTerms did not return expected timeField'
assert (data == terms), 'validateTerms did not return expected terms'