def test_should_tag_single_token_within_partial_prediction_at_smaller_scale(self):
token_1 = SimpleToken(TOKEN_TEXT_1)
structured_document = SimpleStructuredDocument(lines=[SimpleLine([token_1])])
structured_document.set_bounding_box(
structured_document.get_pages()[0],
BoundingBox(0, 0, DEFAULT_WIDTH * 100, DEFAULT_HEIGHT * 100)
)
structured_document.set_bounding_box(
token_1,
BoundingBox(0, 0, DEFAULT_WIDTH * 10, DEFAULT_HEIGHT * 10)
)
annotated_image = filled_image(
BG_COLOR, {TAG_1: COLOR_1},
width=DEFAULT_WIDTH * 10,
height=DEFAULT_HEIGHT * 10
)
fill_rect(
annotated_image,
BoundingBox(0, 0, DEFAULT_WIDTH, DEFAULT_HEIGHT),
COLOR_1
)
annotate_structured_document_using_predicted_images(
structured_document,
[annotated_image]
)
assert structured_document.get_tag(token_1, scope=CV_TAG_SCOPE) == TAG_1
def test_should_not_tag_single_token_not_within_prediction(self):
token_1 = SimpleToken(TOKEN_TEXT_1)
structured_document = SimpleStructuredDocument(
lines=[SimpleLine([token_1])])
structured_document.set_bounding_box(
structured_document.get_pages()[0], DEFAULT_BOUNDING_BOX)
structured_document.set_bounding_box(token_1, DEFAULT_BOUNDING_BOX)
annotate_structured_document_using_predicted_images(
structured_document, [filled_image(BG_COLOR, {TAG_1: COLOR_1})])
assert structured_document.get_tag(token_1, scope=CV_TAG_SCOPE) is None