fix pytest file search

zincware · May 13, 2024 · 3764ecb · 3764ecb
1 parent b7a3c4e
commit 3764ecb
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 4 deletions.
diff --git a/CI/unit_tests/measurements/test_base_measurement.py b/CI/unit_tests/measurements/test_base_measurement.py
@@ -27,7 +27,7 @@
 import numpy as np
 import pytest
 
-from papyrus.measurements.base_measurement import BaseMeasurement
+from papyrus.measurements import BaseMeasurement
 
 
 class DummyMeasurement(BaseMeasurement):

diff --git a/CI/unit_tests/neural_state/test_neural_state.py b/CI/unit_tests/neural_state/test_neural_state.py
@@ -21,7 +21,7 @@
 -------
 """
 
-from papyrus.neural_state.neural_state import NeuralState
+from papyrus.neural_state import NeuralState
 
 
 class TestNeuralState:

diff --git a/CI/unit_tests/neural_state/test_neural_state_creator.py b/CI/unit_tests/neural_state/test_neural_state_creator.py
@@ -23,7 +23,7 @@
 
 import numpy as np
 
-from papyrus.neural_state.neural_state_creator import NeuralStateCreator
+from papyrus.neural_state import NeuralStateCreator
 
 
 class TestNeuralStateCreator:

diff --git a/papyrus/__init__.py b/papyrus/__init__.py
@@ -22,9 +22,10 @@
 papyrus measurements api.
 """
 
-from papyrus import measurements, utils
+from papyrus import measurements, neural_state, utils
 
 __all__ = [
     measurements.__name__,
     utils.__name__,
+    neural_state.__name__,
 ]
diff --git a/pyproject.toml b/pyproject.toml
@@ -15,3 +15,6 @@ keywords = ["neural networks", "artificial intelligence"]
 license = {file = "LICENSE"}
 
 version = "0.0.1"
+
+[tool.pytest.ini_options]
+pythonpath = ["papyrus",]
-Original file line number
+Diff line change
@@ Expand Up / @@ -21,7 +21,7 @@ @@
     -------
     """
-    from papyrus.neural_state.neural_state import NeuralState
+    from papyrus.neural_state import NeuralState
     class TestNeuralState:
@@ Expand Down @@