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