def test_mask_dataset_by_geometry(self): cube = mask_dataset_by_geometry(self.cube, self.triangle) self._assert_clipped_dataset_has_basic_props(cube) cube = mask_dataset_by_geometry(self.cube, self.triangle, save_geometry_wkt=True) self._assert_saved_geometry_wkt_is_fine(cube, 'geometry_wkt') cube = mask_dataset_by_geometry(self.cube, self.triangle, save_geometry_wkt='intersect_geom') self._assert_saved_geometry_wkt_is_fine(cube, 'intersect_geom')
def test_mask_dataset_for_chunked_input(self): cube = chunk_dataset(self.cube, chunk_sizes=dict(time=1, lat=90, lon=90)) cube = mask_dataset_by_geometry(cube, self.triangle) self._assert_clipped_dataset_has_basic_props(cube) self.assertEqual(((1, 1, 1, 1, 1), (4, ), (7, )), cube.temp.chunks) self.assertEqual(((1, 1, 1, 1, 1), (4, ), (7, )), cube.precip.chunks)
def test_mask_dataset_by_geometry_store_mask(self): cube = mask_dataset_by_geometry(self.cube, self.triangle, save_geometry_mask='geom_mask') self._assert_clipped_dataset_has_basic_props(cube) self._assert_dataset_mask_is_fine(cube, 'geom_mask')
def test_mask_dataset_by_geometry_excluded_vars(self): cube = mask_dataset_by_geometry(self.cube, self.triangle, excluded_vars='precip') self._assert_clipped_dataset_has_basic_props(cube)