add imports for classes and modules

bayesian_listener/__init__.py

@@ -5,3 +5,9 @@
 __author__ = """''"""
 __email__ = ''
 __version__ = '#'
+
+from .bayesian_listener import BayesianListener
+
+from . import metrics
+from . import resample
+from . import utils

tests/test_bayesian_listener.py

@@ -1,17 +1,8 @@
 import pytest
-
-
-def test_import_bayesian_listener():
-    try:
-        import bayesian_listener           # noqa
-    except ImportError:
-        pytest.fail('import bayesian_listener failed')
-
-import pytest
 import numpy as np
 from pathlib import Path
 import urllib.request
-from bayesian_listener.bayesian_listener import BayesianListener
+from bayesian_listener import BayesianListener
 from bayesian_listener.coordinates import Coordinates
 
 

tests/test_imports.py

@@ -0,0 +1,41 @@
+import pytest
+
+
+def test_import_bayesian_listener():
+    """Test package import"""
+    try:
+        import bayesian_listener  # noqa
+    except ImportError:
+        pytest.fail('import bayesian_listener failed')
+
+
+def test_import_bayesian_listener_class():
+    """Test BayesianListener class import"""
+    try:
+        from bayesian_listener import BayesianListener  # noqa
+    except ImportError:
+        pytest.fail('from bayesian_listener import BayesianListener failed')
+
+
+def test_import_metrics():
+    """Test import of the metrics module"""
+    try:
+        from bayesian_listener import metrics  # noqa
+    except ImportError:
+        pytest.fail('from bayesian_listener import metrics failed')
+
+
+def test_import_resample():
+    """Test import of the resample module"""
+    try:
+        from bayesian_listener import resample  # noqa
+    except ImportError:
+        pytest.fail('from bayesian_listener import resample failed')
+
+
+def test_import_utils():
+    """Test import of the utils module"""
+    try:
+        from bayesian_listener import utils  # noqa
+    except ImportError:
+        pytest.fail('from bayesian_listener import utils failed')