src/img2pdf_test.py: refactor comparison code to remove code duplication
This commit is contained in:
parent
0ce25d08c2
commit
0e4f0047b2
GPG key ID:
F2CBA5C78FBD83E1
1 changed files with 43 additions and 110 deletions
|
|
@ -284,6 +284,46 @@ def write_png(data, path, bitdepth, colortype, palette=None, iccp=None):
|
||||||
f.write(struct.pack(">I", 0) + block + struct.pack(">I", zlib.crc32(block)))
|
f.write(struct.pack(">I", 0) + block + struct.pack(">I", zlib.crc32(block)))
|
||||||
|
|
||||||
|
|
||||||
|
def compare(im1, im2, exact, icc, cmyk):
|
||||||
|
if exact:
|
||||||
|
if cmyk:
|
||||||
|
raise Exception("cmyk cannot be exact")
|
||||||
|
elif icc:
|
||||||
|
raise Exception("icc cannot be exact")
|
||||||
|
else:
|
||||||
|
subprocess.check_call(
|
||||||
|
[
|
||||||
|
"compare",
|
||||||
|
"-metric",
|
||||||
|
"AE",
|
||||||
|
im1,
|
||||||
|
im2,
|
||||||
|
"null:",
|
||||||
|
]
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
iccargs = []
|
||||||
|
if icc:
|
||||||
|
iccargs = ["-profile", "/usr/share/color/icc/sRGB.icc"]
|
||||||
|
psnr = subprocess.run(
|
||||||
|
["compare"]
|
||||||
|
+ iccargs
|
||||||
|
+ [
|
||||||
|
"-metric",
|
||||||
|
"PSNR",
|
||||||
|
im1,
|
||||||
|
im2,
|
||||||
|
"null:",
|
||||||
|
],
|
||||||
|
check=False,
|
||||||
|
stderr=subprocess.PIPE,
|
||||||
|
).stderr
|
||||||
|
assert psnr != b"0"
|
||||||
|
psnr = float(psnr.strip(b"0"))
|
||||||
|
assert psnr != 0 # or otherwise we would use the exact variant
|
||||||
|
assert psnr > 50
|
||||||
|
|
||||||
|
|
||||||
def compare_ghostscript(tmpdir, img, pdf, gsdevice="png16m", exact=True, icc=False):
|
def compare_ghostscript(tmpdir, img, pdf, gsdevice="png16m", exact=True, icc=False):
|
||||||
if gsdevice in ["png16m", "pnggray"]:
|
if gsdevice in ["png16m", "pnggray"]:
|
||||||
ext = "png"
|
ext = "png"
|
||||||
|
|
@ -303,52 +343,7 @@ def compare_ghostscript(tmpdir, img, pdf, gsdevice="png16m", exact=True, icc=Fal
|
||||||
str(pdf),
|
str(pdf),
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
if exact:
|
compare(str(img), str(tmpdir / "gs-1.") + ext, exact, icc, False)
|
||||||
if icc:
|
|
||||||
subprocess.check_call(
|
|
||||||
[
|
|
||||||
"compare",
|
|
||||||
"-metric",
|
|
||||||
"AE",
|
|
||||||
"(",
|
|
||||||
"-profile",
|
|
||||||
"/usr/share/color/icc/ghostscript/srgb.icc",
|
|
||||||
"-depth",
|
|
||||||
"8",
|
|
||||||
str(img),
|
|
||||||
")",
|
|
||||||
str(tmpdir / "gs-1.") + ext,
|
|
||||||
"null:",
|
|
||||||
]
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
subprocess.check_call(
|
|
||||||
[
|
|
||||||
"compare",
|
|
||||||
"-metric",
|
|
||||||
"AE",
|
|
||||||
str(img),
|
|
||||||
str(tmpdir / "gs-1.") + ext,
|
|
||||||
"null:",
|
|
||||||
]
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
psnr = subprocess.run(
|
|
||||||
[
|
|
||||||
"compare",
|
|
||||||
"-metric",
|
|
||||||
"PSNR",
|
|
||||||
str(img),
|
|
||||||
str(tmpdir / "gs-1.") + ext,
|
|
||||||
"null:",
|
|
||||||
],
|
|
||||||
check=False,
|
|
||||||
stderr=subprocess.PIPE,
|
|
||||||
).stderr
|
|
||||||
assert psnr != b"0"
|
|
||||||
psnr = float(psnr.strip(b"0"))
|
|
||||||
assert psnr != 0 # or otherwise we would use the exact variant
|
|
||||||
assert psnr > 50
|
|
||||||
(tmpdir / ("gs-1." + ext)).unlink()
|
(tmpdir / ("gs-1." + ext)).unlink()
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -356,56 +351,7 @@ def compare_poppler(tmpdir, img, pdf, exact=True, icc=False):
|
||||||
subprocess.check_call(
|
subprocess.check_call(
|
||||||
["pdftocairo", "-r", "96", "-png", str(pdf), str(tmpdir / "poppler")]
|
["pdftocairo", "-r", "96", "-png", str(pdf), str(tmpdir / "poppler")]
|
||||||
)
|
)
|
||||||
if exact:
|
compare(str(img), str(tmpdir / "poppler-1.png"), exact, icc, False)
|
||||||
if icc:
|
|
||||||
raise Exception("not exact with icc")
|
|
||||||
subprocess.check_call(
|
|
||||||
[
|
|
||||||
"compare",
|
|
||||||
"-metric",
|
|
||||||
"AE",
|
|
||||||
str(img),
|
|
||||||
str(tmpdir / "poppler-1.png"),
|
|
||||||
"null:",
|
|
||||||
]
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
if icc:
|
|
||||||
psnr = subprocess.run(
|
|
||||||
[
|
|
||||||
"compare",
|
|
||||||
"-metric",
|
|
||||||
"PSNR",
|
|
||||||
"(",
|
|
||||||
"-profile",
|
|
||||||
"/usr/share/color/icc/ghostscript/srgb.icc",
|
|
||||||
"-depth",
|
|
||||||
"8",
|
|
||||||
str(img),
|
|
||||||
")",
|
|
||||||
str(tmpdir / "poppler-1.png"),
|
|
||||||
"null:",
|
|
||||||
],
|
|
||||||
check=False,
|
|
||||||
stderr=subprocess.PIPE,
|
|
||||||
).stderr
|
|
||||||
else:
|
|
||||||
psnr = subprocess.run(
|
|
||||||
[
|
|
||||||
"compare",
|
|
||||||
"-metric",
|
|
||||||
"PSNR",
|
|
||||||
str(img),
|
|
||||||
str(tmpdir / "poppler-1.png"),
|
|
||||||
"null:",
|
|
||||||
],
|
|
||||||
check=False,
|
|
||||||
stderr=subprocess.PIPE,
|
|
||||||
).stderr
|
|
||||||
assert psnr != b"0"
|
|
||||||
psnr = float(psnr.strip(b"0"))
|
|
||||||
assert psnr != 0 # or otherwise we would use the exact variant
|
|
||||||
assert psnr > 50
|
|
||||||
(tmpdir / "poppler-1.png").unlink()
|
(tmpdir / "poppler-1.png").unlink()
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -422,20 +368,7 @@ def compare_mupdf(tmpdir, img, pdf, exact=True, cmyk=False):
|
||||||
subprocess.check_call(
|
subprocess.check_call(
|
||||||
["mutool", "draw", "-r", "96", "-png", "-o", str(out), str(pdf)]
|
["mutool", "draw", "-r", "96", "-png", "-o", str(out), str(pdf)]
|
||||||
)
|
)
|
||||||
if exact:
|
compare(str(img), str(out), exact, False, cmyk)
|
||||||
if cmyk:
|
|
||||||
raise Exception("cmyk cannot be exact")
|
|
||||||
subprocess.check_call(["compare", "-metric", "AE", str(img), str(out), "null:"])
|
|
||||||
else:
|
|
||||||
psnr = subprocess.run(
|
|
||||||
["compare", "-metric", "PSNR", str(img), str(out), "null:"],
|
|
||||||
check=False,
|
|
||||||
stderr=subprocess.PIPE,
|
|
||||||
).stderr
|
|
||||||
assert psnr != b"0"
|
|
||||||
psnr = float(psnr.strip(b"0"))
|
|
||||||
assert psnr != 0 # or otherwise we would use the exact variant
|
|
||||||
assert psnr > 50
|
|
||||||
out.unlink()
|
out.unlink()
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue