From 5672c878a517804a33eff3ad7abc9d924581219f Mon Sep 17 00:00:00 2001
From: Julien Moura <julien.moura@oslandia.com>
Date: Tue, 18 Jul 2023 15:02:21 +0200
Subject: [PATCH] Move SlabType to enums module

---
 src/rok4/Pyramid.py   | 9 +--------
 src/rok4/enums.py     | 7 +++++++
 tests/test_Pyramid.py | 2 +-
 3 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/src/rok4/Pyramid.py b/src/rok4/Pyramid.py
index 0d176d4..0d7e382 100644
--- 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
index 9a0a486..76300ce 100644
--- 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
index d72ce2f..60576f3 100644
--- 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 *