def test_findSourceValues(): """ @summary: This test checks that source values can be found for proper inputs """ myInstance = SingleTileLCP(M_INPUT_RASTER, '', seaLevelRiseCostFn) myInstance.findSourceCells() assert len(myInstance.sourceCells) > 0
def test_findSourceValues_underSeaLevel(): """ @summary: This test checks that things operate correctly when an input surface is under sea level """ myInstance = SingleTileLCP(BELOW_SEA_LEVEL_RASTER, '', seaLevelRiseCostFn) myInstance.findSourceCells() assert len(myInstance.sourceCells) == 10000
def test_findSourceValues_noSourceValues(): """ @summary: This test checks that things operate correctly when an input surface does not have source values in it """ myInstance = SingleTileLCP(M_INPUT_RASTER_NO_SOURCES, '', seaLevelRiseCostFn) myInstance.findSourceCells() print myInstance.sourceCells assert len(myInstance.sourceCells) == 0
""" import numpy from slr.singleTile.base import SingleTileLCP if __name__ == "__main__": inFn = 'testSurface.asc' outFn = 'testOut.asc' costFn = lambda x,y,z: min(x,y) t1 = SingleTileLCP(inFn, outFn, costFn) t2 = SingleTileLCP(inFn, outFn, costFn) t3 = SingleTileLCP(inFn, outFn, costFn) t1.findSourceCells() v1 = numpy.array([5, 3, 1, 2, 5]) v2 = numpy.array([8, 3, 4, 1, 2, 4]) t2.addSourceVector(v1, 0) t3.addSourceVector(v2, 1) print "t1" print t1.sourceCells print print "t2" print t2.sourceCells print