Fix annotation type check

imi-bigpicture · Feb 21, 2024 · 139a887 · 139a887
1 parent e764c36
commit 139a887
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/tests/test_annotation.py b/tests/test_annotation.py
@@ -678,10 +678,10 @@ def asap_to_geometries(cls, dictionary: Dict[str, Any]) -> List[Geometry]:
     def annotation_to_sectra(
         annotation: Annotation,
     ) -> Tuple[str, List[Dict[str, float]]]:
-        if isinstance(annotation.geometry, Polyline):
-            geometry_type = "Polyline"
-        elif isinstance(annotation.geometry, Polygon):
+        if isinstance(annotation.geometry, Polygon):
             geometry_type = "Area"
+        elif isinstance(annotation.geometry, Polyline):
+            geometry_type = "Polyline"
         else:
             raise NotImplementedError()
         group_coordinates = [
@@ -703,8 +703,8 @@ def sectra_to_geometry(dictionary: Dict[str, Any]) -> Geometry:
     def annotation_to_shapely(annotation: Annotation):
         if isinstance(annotation.geometry, Point):
             return ShapelyPoint(annotation.geometry.to_coords())
-        elif isinstance(annotation.geometry, Polyline):
-            return ShapelyLineString(annotation.geometry.to_coords())
-        elif isinstance(annotation.geometry, Polygon):
+        if isinstance(annotation.geometry, Polygon):
             return ShapelyPolygon(annotation.geometry.to_coords())
+        if isinstance(annotation.geometry, Polyline):
+            return ShapelyLineString(annotation.geometry.to_coords())
         raise NotImplementedError(annotation)