test_get_*: handling when canceling dialog

Author: PierreRaybaut

plotpy/tests/gui/test_get_rectangle.py

@@ -25,11 +25,12 @@ def test_get_rectangle():
         shape = select_with_shape_tool(
             None, RectangleTool, image, "Test", tooldialogclass=PatchedSelectDialog
         )
-        rect = shape.get_rect()
-        if execenv.unattended:
-            assert [round(i) for i in list(rect)] == SEG_AXES_COORDS
-        elif rect is not None:
-            print("Area:", rect)
+        if shape is not None:
+            rect = shape.get_rect()
+            if execenv.unattended:
+                assert [round(i) for i in list(rect)] == SEG_AXES_COORDS
+            elif rect is not None:
+                print("Area:", rect)
 
 
 if __name__ == "__main__":

plotpy/tests/gui/test_get_rectangle_with_svg.py

@@ -43,11 +43,12 @@ def test_get_rectangle_with_svg():
         shape = select_with_shape_tool(
             None, SVGToolExample, image, "Test", tooldialogclass=PatchedSelectDialog
         )
-        rect = shape.get_rect()
-        if execenv.unattended:
-            assert [round(i) for i in list(rect)] == SEG_AXES_COORDS
-        elif rect is not None:
-            print("Area:", rect)
+        if shape is not None:
+            rect = shape.get_rect()
+            if execenv.unattended:
+                assert [round(i) for i in list(rect)] == SEG_AXES_COORDS
+            elif rect is not None:
+                print("Area:", rect)
 
 
 if __name__ == "__main__":

plotpy/tests/gui/test_get_segment.py

@@ -50,12 +50,13 @@ def test_get_segment():
             "Test",
             tooldialogclass=PatchedSelectDialog,
         )
-        rect = shape.get_rect()
-        if execenv.unattended:
-            assert [round(i) for i in list(rect)] == SEG_AXES_COORDS
-        elif rect is not None:
-            distance = np.sqrt((rect[2] - rect[0]) ** 2 + (rect[3] - rect[1]) ** 2)
-            print("Distance:", distance)
+        if shape is not None:
+            rect = shape.get_rect()
+            if execenv.unattended:
+                assert [round(i) for i in list(rect)] == SEG_AXES_COORDS
+            elif rect is not None:
+                distance = np.sqrt((rect[2] - rect[0]) ** 2 + (rect[3] - rect[1]) ** 2)
+                print("Distance:", distance)
 
 
 if __name__ == "__main__":