def get_cdl(self): self.cdl_mask = os.path.join(self.root, 'cdl_mask.tif') if not os.path.isfile(self.cdl_mask): print('get {}'.format(self.cdl_mask)) polygon = self.landsat.get_tile_geometry() cdl = Cdl(year=self.year, target_profile=self.landsat.profile) cdl.get_mask(clip_geometry=polygon, out_file=self.cdl_mask) else: print('{} exists'.format(self.cdl_mask)) self.exclude_rasters.append(self.cdl_mask)
def test_mask(self): l8 = Landsat8(self.dir_name_LC8) shape = 1, l8.rasterio_geometry['height'], l8.rasterio_geometry[ 'width'] polygon = l8.get_tile_geometry() cdl = Cdl(from_file=os.path.join(self.dir_name_LC8, 'cdl.tif')) arr = cdl.get_mask(polygon, out_file=os.path.join(self.dir_name_LC8, 'cdl_mask.tif')) self.assertEqual(shape, arr.shape) self.assertEqual(arr.max(), 1) self.assertEqual(arr.min(), 0)
def get_cdl(self): """download cdl and make a mask, save to the root directory with filename cdl_mask.tif. The cdl is reprojected here. """ self.cdl_mask = os.path.join(self.root, 'cdl_mask.tif') if not os.path.isfile(self.cdl_mask): print('get {}'.format(self.cdl_mask)) polygon = self.landsat.get_tile_geometry() cdl = Cdl(year=self.year, target_profile=self.landsat.profile) cdl.get_mask(clip_geometry=polygon, out_file=self.cdl_mask) else: print('{} exists'.format(self.cdl_mask)) self.exclude_rasters.append(self.cdl_mask)
def test_conforming_array(self): """ Test shape of Gridmet vs. Landsat image. :return: """ l8 = Landsat8(self.dir_name_LC8) shape = 1, l8.rasterio_geometry['height'], l8.rasterio_geometry[ 'width'] polygon = l8.get_tile_geometry() cdl = Cdl(year=self.year, target_profile=l8.profile, out_dir=self.dir_name_LC8) _ = cdl.get_conforming_data(polygon) with rasopen(os.path.join(self.dir_name_LC8, 'cdl.tif')) as dst: arr = dst.read() self.assertEqual(arr.shape, shape)
def test_conform_cdl(self): cdl = Cdl(target_profile=self.target_profile, year=self.year, out_dir=TEST_DATA) cdl.get_original_tif()
def test_download(self): cdl = Cdl(target_profile=self.target_profile, year=self.year) cdl.download_zipped_cdl()