assert invalid gdal datatype raises valueerror

natcap · May 20, 2024 · a332c15 · a332c15
1 parent 77d9ac3
commit a332c15
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/tests/test_geoprocessing.py b/tests/test_geoprocessing.py
@@ -2788,10 +2788,7 @@ def test_create_raster_from_vector_extents_invalid_pixeltype(self):
             self.workspace_dir, 'target_raster.tif')
         target_nodata = -1
         target_pixel_type = gdal.GDT_Int16
-        pygeoprocessing.create_raster_from_vector_extents(
-                source_vector_path, target_raster_path, _DEFAULT_PIXEL_SIZE,
-                target_nodata, target_pixel_type)
-        with self.assertRaises(Exception) as cm:
+        with self.assertRaises(ValueError) as cm:
             pygeoprocessing.create_raster_from_vector_extents(
                 source_vector_path, target_raster_path, _DEFAULT_PIXEL_SIZE,
                 target_nodata, target_pixel_type)