From 96566d70e4f7bc58c9aff010c8e2292c27fe3c80 Mon Sep 17 00:00:00 2001
From: botbikamordehai2-sketch <botbikamordehai2@gmail.com>
Date: Sat, 1 Aug 2026 10:44:37 +0000
Subject: [PATCH] fix: Ensure naive module is available in test_stump.py
 (closes #1160)

---
 tests/test_stump.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/tests/test_stump.py b/tests/test_stump.py
index d5505de96..303468eb2 100644
--- a/tests/test_stump.py
+++ b/tests/test_stump.py
@@ -1,6 +1,8 @@
 import functools
+import sys
+import os
+sys.path.insert(0, os.path.dirname(os.path.dirname(__file__)))  # Add parent directory to path
 
-import naive
 import numpy as np
 import numpy.testing as npt
 import pandas as pd
@@ -10,6 +12,11 @@
 from stumpy import config, rng
 from stumpy.stump import stump
 
+# Ensure naive module is available
+import naive
+if not hasattr(naive, 'stump'):
+    raise ImportError('naive module is not properly initialized. Please run pytest from the repository root or ensure tests/naive is in PYTHONPATH.')
+
 test_data = [
     (
         np.array([9, 8100, -60, 7], dtype=np.float64),