Move SlabType to enums module

rok4 · Jul 18, 2023 · 5672c87 · 5672c87
1 parent 175152b
commit 5672c87
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 9 deletions.
diff --git a/src/rok4/Pyramid.py b/src/rok4/Pyramid.py
@@ -21,14 +21,7 @@
 from rok4.TileMatrixSet import TileMatrixSet, TileMatrix
 from rok4.Storage import *
 from rok4.Utils import *
-from rok4.enums import PyramidType, StorageType
-
-
-class SlabType(Enum):
-    """Slab's type"""
-
-    DATA = "DATA"  # Slab of data, raster or vector
-    MASK = "MASK"  # Slab of mask, only for raster pyramid, image with one band : 0 is nodata, other values are data
+from rok4.enums import PyramidType, SlabType, StorageType
 
 
 ROK4_IMAGE_HEADER_SIZE = 2048

diff --git a/src/rok4/enums.py b/src/rok4/enums.py
@@ -11,6 +11,13 @@ class PyramidType(Enum):
     VECTOR = "VECTOR"
 
 
+class SlabType(Enum):
+    """Slab's type"""
+
+    DATA = "DATA"  # Slab of data, raster or vector
+    MASK = "MASK"  # Slab of mask, only for raster pyramid, image with one band : 0 is nodata, other values are data
+
+
 class StorageType(Enum):
     """Matrice de correspondance entre type de stockage et protocole."""
 

diff --git a/tests/test_Pyramid.py b/tests/test_Pyramid.py
@@ -1,6 +1,6 @@
 from rok4.Pyramid import *
 from rok4.TileMatrixSet import TileMatrixSet
-from rok4.enums import StorageType
+from rok4.enums import SlabType, StorageType
 from rok4.Utils import *
 from rok4.Exceptions import *