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 14:19:06 +02:00
parent caef37376b
commit 1b1f1aafef
1 changed files with 22 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
test/unit_tests/coordinate_transformer_test.py
View File

@ -12,6 +12,11 @@ from image_prediction.transformer.transformers.coordinate.fpdf import FPDFCoordi
from image_prediction.transformer.transformers.coordinate.pdfnet import PDFNetCoordinateTransformer
from test.conftest import array_to_image, add_image
WIDTH = 200
HEIGHT = 100
PAGE_WIDTH = 300
PAGE_HEIGHT = 300
@pytest.mark.parametrize("coordinate_system", ["fpdf"])
def test_fpdf_coordinate_transformer(position_metadata_in_given_system, position_metadata_in_reference_system):
@ -129,10 +134,10 @@ def get_fpdf_corner_metadat():
+--(1,3) +--(3,3)
"""
return {
"top_left": get_metadata_for_coords(0, 0, 100, 100),
"bottom_left": get_metadata_for_coords(0, 200, 100, 300),
"bottom_right": get_metadata_for_coords(200, 200, 300, 300),
"top_right": get_metadata_for_coords(200, 0, 300, 100)
"top_left": get_metadata_for_coords(0, 0, WIDTH, HEIGHT),
"bottom_left": get_metadata_for_coords(0, PAGE_HEIGHT - HEIGHT, WIDTH, PAGE_HEIGHT),
"bottom_right": get_metadata_for_coords(PAGE_WIDTH - WIDTH, PAGE_HEIGHT - HEIGHT, PAGE_WIDTH, PAGE_HEIGHT),
"top_right": get_metadata_for_coords(PAGE_WIDTH - WIDTH, 0, PAGE_WIDTH, HEIGHT)
}
@ -151,10 +156,10 @@ def get_pdfnet_corner_metadata():
(0,0)--+ (2,0)--+
"""
return {
"top_left": get_metadata_for_coords(0, 200, 100, 300),
"bottom_left": get_metadata_for_coords(0, 0, 100, 100),
"bottom_right": get_metadata_for_coords(200, 0, 300, 100),
"top_right": get_metadata_for_coords(200, 200, 300, 300),
"top_left": get_metadata_for_coords(0, PAGE_HEIGHT - HEIGHT, WIDTH, PAGE_HEIGHT),
"bottom_left": get_metadata_for_coords(0, 0, WIDTH, HEIGHT),
"bottom_right": get_metadata_for_coords(PAGE_WIDTH - WIDTH, 0, PAGE_WIDTH, HEIGHT),
"top_right": get_metadata_for_coords(PAGE_WIDTH - WIDTH, PAGE_HEIGHT - HEIGHT, PAGE_WIDTH, PAGE_HEIGHT),
}
@ -164,17 +169,17 @@ def get_metadata_for_coords(x1, y1, x2, y2):
return metadata
def base_position_metadata(width=100, height=100):
def base_position_metadata():
return {
Info.X1: 0,
Info.Y1: 0,
Info.X2: width,
Info.Y2: height,
Info.WIDTH: width,
Info.HEIGHT: height,
Info.X2: WIDTH,
Info.Y2: HEIGHT,
Info.WIDTH: WIDTH,
Info.HEIGHT: HEIGHT,
Info.PAGE_IDX: 0,
Info.PAGE_WIDTH: 300,
Info.PAGE_HEIGHT: 300,
Info.PAGE_WIDTH: PAGE_WIDTH,
Info.PAGE_HEIGHT: PAGE_HEIGHT,
}
@ -187,11 +192,11 @@ def metadata_to_test_image(metadata):
def get_coordinate_test_image():
return array_to_image(np.zeros(shape=(100, 100, 3)))
return array_to_image(np.zeros(shape=(WIDTH, HEIGHT, 3)))
def get_coordinate_test_fpdf():
pdf = fpdf.FPDF(unit="pt", format=(300, 300))
pdf = fpdf.FPDF(unit="pt", format=(PAGE_WIDTH, PAGE_HEIGHT))
return pdf