diff --git a/tests/test_encryption.py b/tests/test_encryption.py index cb2d1f719..f15e072fa 100644 --- a/tests/test_encryption.py +++ b/tests/test_encryption.py @@ -344,6 +344,7 @@ def test_aes_decrypt_corrupted_data(): aes.decrypt(secrets.token_bytes(num)) +@pytest.mark.samples() def test_encrypt_stream_dictionary(pdf_file_path): user_password = secrets.token_urlsafe(10) diff --git a/tests/test_images.py b/tests/test_images.py index 3e9e8a034..b256efe9f 100644 --- a/tests/test_images.py +++ b/tests/test_images.py @@ -73,18 +73,21 @@ def image_similarity( return 1 - mse +@pytest.mark.samples() def test_image_similarity_one(): path_a = SAMPLE_ROOT / "018-base64-image/page-0-QuickPDFImd32aa1ab.png" path_b = path_a assert image_similarity(path_a, path_b) == 1 +@pytest.mark.samples() def test_image_similarity_zero(): path_a = SAMPLE_ROOT / "018-base64-image/page-0-QuickPDFImd32aa1ab.png" path_b = SAMPLE_ROOT / "009-pdflatex-geotopo/page-23-Im2.png" assert image_similarity(path_a, path_b) == 0 +@pytest.mark.samples() def test_image_similarity_mid(): path_a = SAMPLE_ROOT / "018-base64-image/page-0-QuickPDFImd32aa1ab.png" img_b = Image.open(path_a) diff --git a/tests/test_text_extraction.py b/tests/test_text_extraction.py index 58f174f22..1ffa68a3e 100644 --- a/tests/test_text_extraction.py +++ b/tests/test_text_extraction.py @@ -19,6 +19,7 @@ SAMPLE_ROOT = PROJECT_ROOT / "sample-files" +@pytest.mark.samples() @pytest.mark.parametrize(("visitor_text"), [None, lambda a, b, c, d, e: None]) def test_multi_language(visitor_text): reader = PdfReader(RESOURCE_ROOT / "multilang.pdf")