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'