def test_closing_current_polygon(polygon: Polygon):
canvas = PolygonAnnotationCanvas()
canvas.current_polygon = polygon
assume(all((0, 0) < point < canvas.size for point in polygon.points))
assume(not polygon.closed)
assume(len(polygon) > 2)
poly_points = polygon.points.copy()
poly_start = poly_points[0]
closing_click = (
poly_start[0],
min([canvas.size[1], poly_start[1] + (canvas.point_size - 1)]),
)
canvas.on_click(*closing_click)
print(poly_points)
print(closing_click)
assert len(canvas.polygons) == 1
assert canvas.polygons[0].points[:-1] == poly_points
# test current polygon is reset
assert canvas.current_polygon.points == []
def test_editing_mode(polygon: Polygon, polygons: List[Polygon]):
assume(len(polygon) > 2)
canvas = PolygonAnnotationCanvas()
# canvas.polygons = polygons
canvas.current_polygon = polygon
pre_move_points = polygon.points.copy()
canvas.editing = True
# test that dragging is none by default
assert canvas.dragging is None
# clicking on a point sets drag function:
point_to_drag = polygon.points[0]
point_target = (50, 50)
canvas.on_click(*point_to_drag)
assert canvas.dragging is not None
# moving mouse moves the point:
canvas.on_drag(*point_target)
canvas.on_release(*point_target)
# test point has moved
assert canvas.current_polygon.points[0] == (50, 50)
# test everything else stayed put
assert canvas.current_polygon.points[1:] == pre_move_points[1:]
# test release re-set dragging
assert canvas.dragging is None
# test undoing moves the point back
callback = canvas._undo_queue.pop()
callback()
assert canvas.current_polygon.points == pre_move_points