aboutsummaryrefslogtreecommitdiff
path: root/svg-flatten/src/test
diff options
context:
space:
mode:
Diffstat (limited to 'svg-flatten/src/test')
-rw-r--r--svg-flatten/src/test/svg_tests.py60
1 files changed, 49 insertions, 11 deletions
diff --git a/svg-flatten/src/test/svg_tests.py b/svg-flatten/src/test/svg_tests.py
index c457ed4..48ee8fd 100644
--- a/svg-flatten/src/test/svg_tests.py
+++ b/svg-flatten/src/test/svg_tests.py
@@ -1,6 +1,7 @@
#!/usr/bin/env python3
import tempfile
+import shutil
import unittest
from pathlib import Path
import subprocess
@@ -9,7 +10,7 @@ import os
from PIL import Image
import numpy as np
-def run_svg_flatten(input_file, output_file, **kwargs):
+def run_svg_flatten(input_file, output_file, *args, **kwargs):
if 'SVG_FLATTEN' in os.environ:
svg_flatten = os.environ.get('SVG_FLATTEN')
elif (Path(__file__) / '../../build/svg-flatten').is_file():
@@ -19,9 +20,15 @@ def run_svg_flatten(input_file, output_file, **kwargs):
else:
svg_flatten = 'svg-flatten'
- args = [ svg_flatten,
- *(arg for (key, value) in kwargs.items() for arg in (f'--{key.replace("_", "-")}', value)),
- str(input_file), str(output_file) ]
+ args = [ svg_flatten ]
+ for key, value in kwargs.items():
+ key = '--' + key.replace("_", "-")
+ args.append(key)
+
+ if type(value) is not bool:
+ args.append(value)
+ args.append(str(input_file))
+ args.append(str(output_file))
try:
proc = subprocess.run(args, capture_output=True, check=True)
@@ -49,6 +56,11 @@ class SVGRoundTripTests(unittest.TestCase):
# Both are expected and OK.
'stroke_dashes_comparison': 0.03,
'stroke_dashes': 0.05,
+ # The vectorizer tests produce output with lots of edges, which leads to a large amount of aliasing artifacts.
+ 'vectorizer_simple': 0.05,
+ 'vectorizer_clip': 0.05,
+ 'vectorizer_xform': 0.05,
+ 'vectorizer_xform_clip': 0.05,
}
# Force use of rsvg-convert instead of resvg for these test cases
@@ -60,9 +72,21 @@ class SVGRoundTripTests(unittest.TestCase):
'pattern_stroke_dashed'
}
- def compare_images(self, reference, output, test_name, mean, rsvg_workaround=False):
- ref = np.array(Image.open(reference))
- out = np.array(Image.open(output))
+ def compare_images(self, reference, output, test_name, mean, vectorizer_test=False, rsvg_workaround=False):
+ ref, out = Image.open(reference), Image.open(output)
+
+ if vectorizer_test:
+ target_size = (100, 100)
+ ref.thumbnail(target_size, Image.ANTIALIAS)
+ out.thumbnail(target_size, Image.ANTIALIAS)
+ ref, out = np.array(ref), np.array(out)
+
+ else:
+ ref, out = np.array(ref), np.array(out)
+
+ ref, out = ref.astype(float).mean(axis=2), out.astype(float).mean(axis=2)
+
+
if rsvg_workaround:
# For some stupid reason, rsvg-convert does not actually output black as in "black" pixels when asked to.
# Instead, it outputs #010101. We fix this in post here.
@@ -86,9 +110,24 @@ class SVGRoundTripTests(unittest.TestCase):
tempfile.NamedTemporaryFile(suffix='.png') as tmp_out_png,\
tempfile.NamedTemporaryFile(suffix='.png') as tmp_in_png:
- run_svg_flatten(test_in_svg, tmp_out_svg.name, format='svg')
-
use_rsvg = test_in_svg.stem in SVGRoundTripTests.rsvg_override
+ vectorizer_test = test_in_svg.stem.startswith('vectorizer')
+ contours_test = test_in_svg.stem.startswith('contours')
+
+ if not vectorizer_test:
+ run_svg_flatten(test_in_svg, tmp_out_svg.name, format='svg')
+
+ else:
+ run_svg_flatten(test_in_svg, tmp_out_svg.name, format='svg',
+ svg_white_is_gerber_dark=True,
+ clear_color='black', dark_color='white')
+
+ if contours_test:
+ run_svg_flatten(test_in_svg, tmp_out_svg.name,
+ clear_color='black', dark_color='white',
+ svg_white_is_gerber_dark=True,
+ format='svg',
+ vectorizer='binary-contours')
if not use_rsvg: # default!
subprocess.run(['resvg', tmp_out_svg.name, tmp_out_png.name], check=True, stdout=subprocess.DEVNULL)
@@ -101,10 +140,9 @@ class SVGRoundTripTests(unittest.TestCase):
try:
self.compare_images(tmp_in_png, tmp_out_png, test_in_svg.stem,
SVGRoundTripTests.test_mean_overrides.get(test_in_svg.stem, SVGRoundTripTests.test_mean_default),
- rsvg_workaround=use_rsvg)
+ vectorizer_test, rsvg_workaround=use_rsvg)
except AssertionError as e:
- import shutil
shutil.copyfile(tmp_in_png.name, f'/tmp/gerbolyze-fail-{test_in_svg.stem}-in.png')
shutil.copyfile(tmp_out_png.name, f'/tmp/gerbolyze-fail-{test_in_svg.stem}-out.png')
foo = list(e.args)