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)