def setup(self): self.workspace = os.path.abspath(resource_filename('tidegates.testing', 'cleanup_temp_results')) self.template_file = resource_filename('tidegates.testing.cleanup_temp_results', 'test_dem.tif') self.template = utils.load_data(self.template_file, 'raster') raster1 = utils.array_to_raster(numpy.random.normal(size=(30, 30)), self.template) raster2 = utils.array_to_raster(numpy.random.normal(size=(60, 60)), self.template) self.name1 = 'temp_1.tif' self.name2 = 'temp_2.tif' self.path1 = os.path.join(self.workspace, self.name1) self.path2 = os.path.join(self.workspace, self.name2) with utils.OverwriteState(True), utils.WorkSpace(self.workspace): raster1.save(self.path1) raster2.save(self.path2)
def test_array_to_raster(): template_file = resource_filename("tidegates.testing.array_to_raster", 'test_raster2') template = arcpy.Raster(template_file) array = numpy.arange(5, 25).reshape(4, 5).astype(float) raster = utils.array_to_raster(array, template) nt.assert_true(isinstance(raster, arcpy.Raster)) nt.assert_true(raster.extent.equals(template.extent)) nt.assert_equal(raster.meanCellWidth, template.meanCellWidth) nt.assert_equal(raster.meanCellHeight, template.meanCellHeight)
def setup(self): self.workspace = os.path.abspath( resource_filename('tidegates.testing', 'cleanup_temp_results')) self.template_file = resource_filename( 'tidegates.testing.cleanup_temp_results', 'test_dem.tif') self.template = utils.load_data(self.template_file, 'raster') raster1 = utils.array_to_raster(numpy.random.normal(size=(30, 30)), self.template) raster2 = utils.array_to_raster(numpy.random.normal(size=(60, 60)), self.template) self.name1 = 'temp_1.tif' self.name2 = 'temp_2.tif' self.path1 = os.path.join(self.workspace, self.name1) self.path2 = os.path.join(self.workspace, self.name2) with utils.OverwriteState(True), utils.WorkSpace(self.workspace): raster1.save(self.path1) raster2.save(self.path2)