diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/astra/__init__.py | 4 | ||||
| -rw-r--r-- | python/astra/tests.py | 10 |
2 files changed, 12 insertions, 2 deletions
diff --git a/python/astra/__init__.py b/python/astra/__init__.py index 4675229..e813194 100644 --- a/python/astra/__init__.py +++ b/python/astra/__init__.py @@ -27,7 +27,7 @@ from . import matlab as m from .creators import astra_dict,create_vol_geom, create_proj_geom, create_backprojection, create_sino, create_reconstruction, create_projector,create_sino3d_gpu, create_backprojection3d_gpu from .functions import data_op, add_noise_to_sino, clear, move_vol_geom from .extrautils import clipCircle -from .astra import set_gpu_index, get_gpu_info +from .astra import set_gpu_index, get_gpu_info, use_cuda from . import data2d from . import astra from . import data3d @@ -39,7 +39,7 @@ from . import plugin from . import plugins from . import log from .optomo import OpTomo -from .tests import test_noCUDA, test_CUDA +from .tests import test, test_noCUDA, test_CUDA __version__ = '1.9.0dev' diff --git a/python/astra/tests.py b/python/astra/tests.py index 32afd36..f7c2c04 100644 --- a/python/astra/tests.py +++ b/python/astra/tests.py @@ -95,3 +95,13 @@ def test_CUDA(): ok3 = _basic_par3d_cuda() if not (ok1 and ok2 and ok3): raise RuntimeError("Test failed") + +def test(): + """Perform a very basic functionality test""" + + import astra + if astra.use_cuda(): + test_CUDA() + else: + print("No GPU support available") + test_noCUDA() |