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

Refactoring

Browse Source 
Add fixture for page partitioner
This commit is contained in:
Matthias Bisping 2023-02-01 16:29:03 +01:00
parent 2fb450943e
commit b3cf3e4454
1 changed files with 13 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
test/fixtures/page_generation/page.py vendored
View File

@ -33,7 +33,7 @@ logger.remove()
logger.add(sys.stderr, level="INFO") logger.add(sys.stderr, level="INFO")
random_seed = random.randint(0, 2**32 - 1) random_seed = random.randint(0, 2**32 - 1)
random_seed = 2973413116 # random_seed = 2973413116
# random_seed = 2212357755 # random_seed = 2212357755
rnd = random.Random(random_seed) rnd = random.Random(random_seed)
@ -1307,20 +1307,21 @@ def drop_small_boxes(boxes: Iterable[Rectangle], page_width, page_height, min_pe
return lremove(small, boxes) return lremove(small, boxes)
@pytest.fixture(
params=[
TwoColumnPagePartitioner,
# RandomPagePartitioner
]
)
def page_partitioner(request):
return request.param()
@pytest.fixture @pytest.fixture
def page_with_opaque_content( def page_with_opaque_content(
blank_page, blank_page, tinted_blank_page, texture, texture_fn, page_partitioner
tinted_blank_page,
texture,
texture_fn,
) -> Tuple[np.ndarray, Iterable[Rectangle]]: ) -> Tuple[np.ndarray, Iterable[Rectangle]]:
"""Creates a page with content""" """Creates a page with content"""
page_partitioner = rnd.choice(
[
TwoColumnPagePartitioner(),
# RandomPagePartitioner(),
]
)
texture = random_flip(texture) texture = random_flip(texture)
texture = texture_fn(texture) texture = texture_fn(texture)
@ -1336,15 +1337,9 @@ def page_with_opaque_content(
@pytest.fixture @pytest.fixture
def page_with_translucent_content( def page_with_translucent_content(
blank_page, tinted_blank_page, texture, texture_fn blank_page, tinted_blank_page, texture, texture_fn, page_partitioner
) -> Tuple[np.ndarray, List[Rectangle]]: ) -> Tuple[np.ndarray, List[Rectangle]]:
"""Creates a page with content""" """Creates a page with content"""
page_partitioner = rnd.choice(
[
TwoColumnPagePartitioner(),
# RandomPagePartitioner(),
]
)
boxes = page_partitioner(blank_page) boxes = page_partitioner(blank_page)
content_generator = ContentGenerator() content_generator = ContentGenerator()