Marmelator/image-classification-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 29 Wiki Activity

refactoring

Browse Source
This commit is contained in:
Matthias Bisping 2022-04-04 10:23:22 +02:00
parent b4b0058475
commit 0976971117
1 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
test/unit_tests/coordinate_transformer_test.py
View File

@ -28,6 +28,10 @@ def corner(request):
return request.param return request.param
def set_coords(metadata, x1, y1, x2, y2):
metadata.update({Info.X1: x1, Info.Y1: y1, Info.X2: x2, Info.Y2: y2})
def get_fpdf_coordinates(corner): def get_fpdf_coordinates(corner):
"""Origin top left, y1 <= y2; all coords on page are positive """Origin top left, y1 <= y2; all coords on page are positive
(0,0)--+--(2,0)--+ (0,0)--+--(2,0)--+
@ -41,16 +45,16 @@ def get_fpdf_coordinates(corner):
metadata = base_position_metadata() metadata = base_position_metadata()
if corner == "top_left": if corner == "top_left":
metadata.update({Info.X1: 0, Info.Y1: 0, Info.X2: 100, Info.Y2: 100}) set_coords(metadata, 0, 0, 100, 100)
elif corner == "bottom_left": elif corner == "bottom_left":
metadata.update({Info.X1: 0, Info.Y1: 200, Info.X2: 100, Info.Y2: 300}) set_coords(metadata, 0, 200, 100, 300)
elif corner == "bottom_right": elif corner == "bottom_right":
metadata.update({Info.X1: 200, Info.Y1: 200, Info.X2: 300, Info.Y2: 300}) set_coords(metadata, 200, 200, 300, 300)
elif corner == "top_right": elif corner == "top_right":
metadata.update({Info.X1: 200, Info.Y1: 0, Info.X2: 300, Info.Y2: 100}) set_coords(metadata, 200, 0, 300, 100)
else: else:
raise ValueError(f"No coordinates specified for corner {corner}.") raise ValueError(f"No coordinates specified for corner {corner}.")
@ -75,16 +79,16 @@ def get_pdfnet_coordinates(corner):
metadata = base_position_metadata() metadata = base_position_metadata()
if corner == "top_left": if corner == "top_left":
metadata.update({Info.X1: 0, Info.Y1: 200, Info.X2: 100, Info.Y2: 300}) set_coords(metadata, 0, 200, 100, 300)
elif corner == "bottom_left": elif corner == "bottom_left":
metadata.update({Info.X1: 0, Info.Y1: 0, Info.X2: 100, Info.Y2: 100}) set_coords(metadata, 0, 0, 100, 100)
elif corner == "bottom_right": elif corner == "bottom_right":
metadata.update({Info.X1: 200, Info.Y1: 0, Info.X2: 300, Info.Y2: 100}) set_coords(metadata, 200, 0, 300, 100)
elif corner == "top_right": elif corner == "top_right":
metadata.update({Info.X1: 200, Info.Y1: 200, Info.X2: 300, Info.Y2: 300}) set_coords(metadata, 200, 200, 300, 300)
else: else:
raise ValueError(f"No coordinates specified for corner {corner}.") raise ValueError(f"No coordinates specified for corner {corner}.")