Refactor: use enum for contour shapes in contour application tests

Author: PierreRaybaut

datalab/tests/features/image/contour_app_test.py

@@ -7,6 +7,7 @@
 # pylint: disable=invalid-name  # Allows short reference names like x, y, ...
 # guitest: show
 
+import sigima.enums
 import sigima.params
 from sigima.objects import NewImageParam
 from sigima.tests.data import create_multigaussian_image
@@ -21,7 +22,7 @@ def test_contour_app():
     newparam = NewImageParam.create(height=200, width=200)
     with datalab_test_app_context() as win:
         panel = win.imagepanel
-        for shape in ("polygon", "circle", "ellipse"):
+        for shape in sigima.enums.ContourShape:
             ima1 = create_multigaussian_image(newparam)
             ima1.set_metadata_option("colormap", "gray")
             panel.add_object(ima1)

datalab/tests/features/image/contour_fabryperot_app_test.py

@@ -7,6 +7,7 @@
 # pylint: disable=invalid-name  # Allows short reference names like x, y, ...
 # guitest: show
 
+import sigima.enums
 import sigima.objects
 import sigima.params
 from sigima.tests.data import get_test_image
@@ -19,7 +20,7 @@ def test_contour_app() -> None:
     with datalab_test_app_context() as win:
         panel = win.imagepanel
 
-        shape = "circle"
+        shape = sigima.enums.ContourShape.CIRCLE
 
         ima1 = get_test_image("fabry-perot1.jpg")
         ima1.set_metadata_option("colormap", "gray")