Add guard against missing shapely in unit tests

Argmaster · Jan 29, 2025 · b912cfc · b912cfc
1 parent 347b5d7
commit b912cfc
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/test/unit/test_gerber/test_console/test_gerber.py b/test/unit/test_gerber/test_console/test_gerber.py
@@ -6,8 +6,9 @@
 from click.testing import CliRunner
 from PIL import Image
 
-from pygerber.console.gerber import bmp, format_cmd, jpeg, png, svg, tiff, webp
+from pygerber.console.gerber import bmp, format_cmd, jpeg, png, tiff, webp
 from pygerber.examples import ExamplesEnum, get_example_path
+from pygerber.vm.shapely.vm import is_shapely_available
 from test.assets.assetlib import ImageAnalyzer, SvgImageAsset
 from test.assets.reference.pygerber.console.gerber import (
     CONVERT_BMP_REFERENCE_IMAGE,
@@ -25,6 +26,10 @@
 MIN_PNG_SSIM = 0.99
 
 
+if is_shapely_available():
+    from pygerber.console.gerber import svg
+
+
 @tag(Tag.PILLOW, Tag.OPENCV, Tag.SKIMAGE)
 def test_gerber_convert_png(*, is_regeneration_enabled: bool) -> None:
     runner = CliRunner()