fix: support direct pytest package import

2026-08-16 03:51:44 +02:00
parent 34842561a6
commit 2477c2e774
1 changed files with 6 additions and 2 deletions
@@ -1,5 +1,9 @@
-from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
-from .spectrum_h3_compat import install_spectrum_h3_compat
+if __package__:
+    from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
+    from .spectrum_h3_compat import install_spectrum_h3_compat
+else:  # pragma: no cover - direct pytest/import fallback
+    from nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
+    from spectrum_h3_compat import install_spectrum_h3_compat

 install_spectrum_h3_compat()