Refactor from ImageInfo index to attribute accessing

tests/test_main.py

@@ -12,6 +12,7 @@ from ocrmypdf import leptonica
 from ocrmypdf.pdfa import file_claims_pdfa
 from ocrmypdf.exec import ghostscript
 import logging
+from math import isclose
 
 
 check_ocrmypdf = pytest.helpers.check_ocrmypdf
@@ -629,7 +630,7 @@ def test_jbig2_passthrough(spoof_tesseract_cache, resources, outpdf):
         env=spoof_tesseract_cache)
 
     out_pageinfo = PdfInfo(out)
-    assert out_pageinfo[0].images[0]['enc'] == 'jbig2'
+    assert out_pageinfo[0].images[0].enc == 'jbig2'
 
 
 def test_stdin(spoof_tesseract_noop, ocrmypdf_exec, resources, outpdf):
@@ -746,8 +747,8 @@ def test_very_high_dpi(spoof_tesseract_cache, resources, outpdf):
     pdfinfo = PdfInfo(outpdf)
 
     image = pdfinfo[0].images[0]
-    assert image['dpi_w'] == image['dpi_h']
-    assert image['dpi_w'] == 2400
+    assert isclose(image.xres, image.yres)
+    assert isclose(image.xres, 2400)
 
 
 def test_overlay(spoof_tesseract_noop, resources, outpdf):
@@ -905,16 +906,16 @@ def test_compression_preserved(spoof_tesseract_noop, ocrmypdf_exec,
     pdfimage = pdfinfo[0].images[0]
 
     if input_file.endswith('.png'):
-        assert pdfimage['enc'] != 'jpeg', \
+        assert pdfimage.enc != 'jpeg', \
             "Lossless compression changed to lossy!"
     elif input_file.endswith('.jpg'):
-        assert pdfimage['enc'] == 'jpeg', \
+        assert pdfimage.enc == 'jpeg', \
             "Lossy compression changed to lossless!"
     if im.mode.startswith('RGB') or im.mode.startswith('BGR'):
-        assert pdfimage['color'] == 'rgb', \
+        assert pdfimage.color == 'rgb', \
             "Colorspace changed"
     elif im.mode.startswith('L'):
-        assert pdfimage['color'] == 'gray', \
+        assert pdfimage.color == 'gray', \
             "Colorspace changed"
 
 
@@ -950,15 +951,15 @@ def test_compression_changed(spoof_tesseract_noop, ocrmypdf_exec,
     pdfimage = pdfinfo[0].images[0]
 
     if compression == 'jpeg':
-        assert pdfimage['enc'] == 'jpeg'
+        assert pdfimage.enc == 'jpeg'
     elif compression == 'lossless':
-        assert pdfimage['enc'] == 'image'
+        assert pdfimage.enc == 'image'
 
     if im.mode.startswith('RGB') or im.mode.startswith('BGR'):
-        assert pdfimage['color'] == 'rgb', \
+        assert pdfimage.color == 'rgb', \
             "Colorspace changed"
     elif im.mode.startswith('L'):
-        assert pdfimage['color'] == 'gray', \
+        assert pdfimage.color == 'gray', \
             "Colorspace changed"
 
 

tests/test_pageinfo.py

@@ -5,6 +5,7 @@ from ocrmypdf import pageinfo
 from reportlab.pdfgen.canvas import Canvas
 from PIL import Image
 from tempfile import NamedTemporaryFile
+from math import isclose
 from contextlib import suppress
 import os
 import shutil
@@ -62,12 +63,12 @@ def test_single_page_image(outdir):
     assert len(page.images) == 1
 
     pdfimage = page.images[0]
-    assert pdfimage['width'] == 8
-    assert pdfimage['color'] == 'gray'
+    assert pdfimage.width == 8
+    assert pdfimage.color == 'gray'
 
     # DPI in a 1"x1" is the image width
-    assert abs(pdfimage['dpi_w'] - 8) < 1e-5
-    assert abs(pdfimage['dpi_h'] - 8) < 1e-5
+    assert isclose(pdfimage.xres, 8)
+    assert isclose(pdfimage.yres, 8)
 
 
 def test_single_page_inline_image(outdir):
@@ -86,9 +87,9 @@ def test_single_page_inline_image(outdir):
     pdfinfo = pageinfo.PdfInfo(filename)
     print(pdfinfo)
     pdfimage = pdfinfo[0].images[0]
-    assert (pdfimage['dpi_w'] - 8) < 1e-5
-    assert pdfimage['color'] != '-'
-    assert pdfimage['width'] == 8
+    assert isclose(pdfimage.xres, 8)
+    assert pdfimage.color != '-'
+    assert pdfimage.width == 8
 
 
 def test_jpeg(resources, outdir):
@@ -97,8 +98,8 @@ def test_jpeg(resources, outdir):
     pdfinfo = pageinfo.PdfInfo(filename)
 
     pdfimage = pdfinfo[0].images[0]
-    assert pdfimage['enc'] == 'jpeg'
-    assert (pdfimage['dpi_w'] - 150) < 1e-5
+    assert pdfimage.enc == 'jpeg'
+    assert isclose(pdfimage.xres, 150)
 
 
 def test_form_xobject(resources):
@@ -106,7 +107,7 @@ def test_form_xobject(resources):
 
     pdfinfo = pageinfo.PdfInfo(filename)
     pdfimage = pdfinfo[0].images[0]
-    assert pdfimage['width'] == 50
+    assert pdfimage.width == 50
 
 
 def test_no_contents(resources):

tests/test_tess4.py

@@ -117,4 +117,4 @@ def test_content_preservation(ensure_tess4, resources, outpdf):
 
     info = pageinfo.PdfInfo(outpdf)
     page = info[0]
-    assert len(page['images']) > 1, "masked were rasterized"
+    assert len(page.images) > 1, "masked were rasterized"