test: fix imports

scikit-hep · Nov 9, 2024 · 0971dd4 · 0971dd4
1 parent 5ef1bc4
commit 0971dd4
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -39,7 +39,7 @@ jobs:
             python-version: "3.9"
           - os: macos-13
             python-version: "3.9"
-          - os: macos-14
+          - os: macos-latest
             python-version: "3.12"  # old versions not supported
     name: Check Python ${{ matrix.python-version }} ${{ matrix.os }}
     steps:

diff --git a/tests/modeling/test_bayesianblocks.py b/tests/modeling/test_bayesianblocks.py
@@ -1,9 +1,11 @@
-import os
+from pathlib import Path
+
 import numpy as np
+
 import hepstats
 from hepstats.modeling import bayesian_blocks
 
-answer_dir = os.path.dirname(hepstats.__file__) + "/../../tests/modeling/data"
+answer_dir = Path(hepstats.__file__) / "data"
 
 
 def test_bayesian_blocks(cmdopt, data_gen):
@@ -12,10 +14,10 @@ def test_bayesian_blocks(cmdopt, data_gen):
     be3 = bayesian_blocks(data_gen[0], weights=data_gen[2])
 
     if cmdopt == "generate":
-        with open(answer_dir + "/answers_bayesian_blocks.npz", "wb") as f:
+        with open(answer_dir / "answers_bayesian_blocks.npz", "wb") as f:
             np.savez(f, be1=be1, be2=be2, be3=be3)
     elif cmdopt == "test":
-        answers = np.load(answer_dir + "/answers_bayesian_blocks.npz")
+        answers = np.load(answer_dir / "answers_bayesian_blocks.npz")
         assert np.all(be1 == answers["be1"])
         assert np.all(be2 == answers["be2"])
         assert np.all(be3 == answers["be3"])