def test_given_image_with_text_and_matrix_fill_then_text_is_colored_out():
# Image with PII entities
image = get_resource_image("ocr_test.png")
redacted_image = ImageRedactorEngine().redact(image, red_fill)
expected_result_image = get_resource_image("ocr_test_redacted_matrix.png")
assert compare_images(redacted_image, expected_result_image)
assert not compare_images(redacted_image, image)
def test_given_image_then_text_entities_are_recognized_correctly():
# Image with PII entities
image = get_resource_image("ocr_test.png")
analyzer_results = ImageAnalyzerEngine().analyze(image)
assert len(analyzer_results) == 7
results = __get_expected_ocr_test_image_analysis_results()
for i in range(7):
assert analyzer_results[i] == results[i]
def test_given_image_without_text_then_no_entities_recognized():
# Image without PII entities
image = get_resource_image("no_ocr.jpg")
analyzer_results = ImageAnalyzerEngine().analyze(image)
assert len(analyzer_results) == 0
def test_given_image_with_text_and_fill_then_text_is_greyed_out():
# Image with PII entities
image = get_resource_image("ocr_test.png")
result_image = get_resource_image("ocr_test_redacted.png")
redacted_image = ImageRedactorEngine().redact(image, 1)
assert compare_images(redacted_image, result_image)
def test_given_image_without_text_and_fill_then_image_does_not_change():
# Image without PII entities
image = get_resource_image("no_ocr.jpg")
redacted_image = ImageRedactorEngine().redact(image, red_fill)
assert compare_images(redacted_image, image)
def test_given_image_without_text_and_pii_verify_then_image_does_not_change():
# Image without PII entities
image = get_resource_image("no_ocr.png")
pii_verifying_image = ImagePiiVerifyEngine().verify(image)
assert compare_images(pii_verifying_image, image)