fix tests

dbbs-lab · Mar 28, 2024 · 4a1c80f · 4a1c80f
1 parent f0525a1
commit 4a1c80f
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 6 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -33,6 +33,7 @@ jobs:
           pip install wheel
           pip install -r requirements.txt --prefer-binary
           pip install -e . --no-deps
+          pip install bsb-test==4.0.0rc3 --no-deps
       - name: Run tests & coverage
         run: |
           coverage run -p -m unittest

diff --git a/test/test_chunks.py b/test/test_chunks.py
@@ -1,8 +1,7 @@
 import unittest
 
 import numpy as np
-from bsb.core import Scaffold
-from bsb.storage import Chunk
+from bsb import Chunk, Scaffold
 from bsb_test import NumpyTestCase, RandomStorageFixture, skip_parallel, timeout
 from bsb_test.configs import get_test_config
 

diff --git a/test/test_cs.py b/test/test_cs.py
@@ -1,7 +1,6 @@
 import unittest
 
-from bsb.core import Scaffold
-from bsb.services.pool import WorkflowError
+from bsb import Scaffold, WorkflowError
 from bsb_test import FixedPosConfigFixture, NumpyTestCase, RandomStorageFixture
 
 from bsb_hdf5.connectivity_set import LocationOutOfBoundsError

diff --git a/test/test_mr.py b/test/test_mr.py
@@ -4,8 +4,7 @@
 
 import h5py
 import numpy as np
-from bsb.services import MPI
-from bsb.storage import Storage
+from bsb import MPI, Storage
 
 
 class TestHandcrafted(unittest.TestCase):