From 2477c2e7749b7fdb1ae142f2bd152f0678a2a484 Mon Sep 17 00:00:00 2001
From: xmarre <54859656+xmarre@users.noreply.github.com>
Date: Sun, 16 Aug 2026 03:51:44 +0200
Subject: [PATCH] fix: support direct pytest package import

---
 __init__.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/__init__.py b/__init__.py
index 273cd43..8c5bcf5 100644
--- a/__init__.py
+++ b/__init__.py
@@ -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()