Marmelator/cv-analysis-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 37 Wiki Activity

Refactoring

Browse Source
This commit is contained in:
Matthias Bisping 2023-01-04 18:29:52 +01:00
parent cd5457840b
commit dfd87cb4b0
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
test/unit_tests/figure_detection/figure_detection_test.py
View File

@ -3,6 +3,7 @@ from math import prod
import cv2
import pytest
from cv_analysis.utils.spacial import area
from test.utils.utils import powerset
@ -26,7 +27,7 @@ class TestFindPrimaryTextRegions:
@pytest.mark.parametrize("error_tolerance", [0.025])
def test_page_with_only_text_yields_no_figures(self, figure_detection_pipeline, page_with_text, error_tolerance):
result_rectangles = figure_detection_pipeline(page_with_text)
result_figure_areas = sum(map(lambda r: (r.width * r.height), result_rectangles))
result_figure_areas = sum(map(area, result_rectangles))
page_area = prod(page_with_text.shape)
error = result_figure_areas / page_area
@ -46,7 +47,7 @@ class TestFindPrimaryTextRegions:
):
result_rectangles = list(figure_detection_pipeline(page_with_images_and_text))
result_figure_areas = sum(map(lambda r: (r.width * r.height), result_rectangles))
result_figure_areas = sum(map(area, result_rectangles))
expected_figure_area = prod(image_size)
error = abs(result_figure_areas - expected_figure_area) / expected_figure_area