diff --git a/bayesian_listener/__init__.py b/bayesian_listener/__init__.py index 6e048e1..5a8aef7 100644 --- a/bayesian_listener/__init__.py +++ b/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 diff --git a/tests/test_bayesian_listener.py b/tests/test_bayesian_listener.py index 57fd5f4..8e90b21 100644 --- a/tests/test_bayesian_listener.py +++ b/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 diff --git a/tests/test_imports.py b/tests/test_imports.py new file mode 100644 index 0000000..1ddeabc --- /dev/null +++ b/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')