enthought · prabhuramachandran · Apr 1, 2016 · Mar 30, 2016 · Mar 30, 2016
diff --git a/mayavi/tests/runtests.py b/mayavi/tests/runtests.py
@@ -60,11 +60,17 @@ def find_tests(tests):
             try:
                 # Import the module
                 components = test.split('.')
-                modname = '.'.join(components[:-1])
-                symbol = components[-1]
-                mod = __import__(modname, globals(), locals(), [symbol])
-                s = getattr(mod, symbol)
-                d = dirname(s.__file__)
+                if len(components) > 1:
+                    modname = '.'.join(components[:-1])
+                    symbol = components[-1]
+                    mod = __import__(modname, globals(), locals(), [symbol])
+                    s = getattr(mod, symbol)
+                    d = dirname(s.__file__)
+                else:
+                    modname = components[0]
+                    mod = __import__(modname, globals(), locals(), [])
+                    d = dirname(mod.__file__)
+
                 files.extend(get_tests_in_dir(d))
             except ImportError:
                 msg = 'Warning: %s is neither a file/directory or '\

diff --git a/mayavi/tests/test_contour.py b/mayavi/tests/test_contour.py
@@ -19,7 +19,7 @@
 from mayavi.modules.contour_grid_plane import ContourGridPlane
 from mayavi.modules.scalar_cut_plane import ScalarCutPlane
 
-from . import datasets
+from mayavi.tests import datasets
 
 class TestContour(unittest.TestCase):
 

diff --git a/mayavi/tests/test_grid_plane.py b/mayavi/tests/test_grid_plane.py
@@ -8,7 +8,7 @@
 from io import BytesIO
 import copy
 import unittest
-from . import datasets
+from mayavi.tests import datasets
 
 # Local imports.
 from mayavi.core.engine import Engine

diff --git a/mayavi/tests/test_image_data_probe.py b/mayavi/tests/test_image_data_probe.py
@@ -10,7 +10,7 @@
 import unittest
 
 # Local imports.
-from .common import get_example_data
+from mayavi.tests.common import get_example_data
 
 # Enthought library imports
 from mayavi.core.null_engine import NullEngine

diff --git a/mayavi/tests/test_image_data_reader.py b/mayavi/tests/test_image_data_reader.py
@@ -7,7 +7,7 @@
 import numpy
 
 # Local imports.
-from .common import get_example_data
+from mayavi.tests.common import get_example_data
 
 # Enthought library imports
 from mayavi.sources.image_reader import ImageReader

diff --git a/mayavi/tests/test_mlab_scene_model.py b/mayavi/tests/test_mlab_scene_model.py
@@ -9,7 +9,7 @@
 from mayavi.tools.mlab_scene_model import MlabSceneModel
 from mayavi import mlab
 
-from .test_mlab_integration import TestMlabNullEngine
+from mayavi.tests.test_mlab_integration import TestMlabNullEngine
 
 ###############################################################################
 # class `TestMlabSceneModel`

diff --git a/mayavi/tests/test_optional_collection.py b/mayavi/tests/test_optional_collection.py
@@ -19,7 +19,7 @@
 from mayavi.modules.api import Surface
 from mayavi.sources.vtk_data_source import VTKDataSource
 
-from . import datasets
+from mayavi.tests import datasets
 
 class TestOptionalCollection(unittest.TestCase):
 

diff --git a/mayavi/tests/test_plot3d_mb_reader.py b/mayavi/tests/test_plot3d_mb_reader.py
@@ -10,7 +10,7 @@
 import unittest
 
 # Local imports.
-from .common import get_example_data
+from mayavi.tests.common import get_example_data
 
 # Enthought library imports
 from mayavi.core.null_engine import NullEngine

diff --git a/mayavi/tests/test_poly_data_reader.py b/mayavi/tests/test_poly_data_reader.py
@@ -6,7 +6,7 @@
 import unittest
 
 # Local imports.
-from .common import get_example_data
+from mayavi.tests.common import get_example_data
 
 # Enthought library imports
 from mayavi.sources.poly_data_reader import PolyDataReader

diff --git a/mayavi/tests/test_registry.py b/mayavi/tests/test_registry.py
@@ -13,7 +13,7 @@
 from mayavi.core.pipeline_info import PipelineInfo
 
 # Local Imports
-from .common import get_example_data
+from mayavi.tests.common import get_example_data
 
 class DummyReader(PLOT3DReader):
 

diff --git a/mayavi/tests/test_set_active_attribute.py b/mayavi/tests/test_set_active_attribute.py
@@ -18,7 +18,7 @@
 from mayavi.modules.api import Surface, Outline
 
 # Local imports.
-from .common import get_example_data
+from mayavi.tests.common import get_example_data
 
 class TestSetActiveAttribute(unittest.TestCase):
 

diff --git a/mayavi/tests/test_unstructured_data_reader.py b/mayavi/tests/test_unstructured_data_reader.py
@@ -11,7 +11,7 @@
 from mayavi.tests.data_reader_test_base import DataReaderTestBase
 
 # Local imports.
-from .common import get_example_data
+from mayavi.tests.common import get_example_data
 
 old_pipeline = is_old_pipeline()
 

diff --git a/mayavi/tests/test_vtk_data_source.py b/mayavi/tests/test_vtk_data_source.py
@@ -19,7 +19,7 @@
 from mayavi.modules.scalar_cut_plane import ScalarCutPlane
 from tvtk.api import tvtk
 
-from . import datasets
+from mayavi.tests import datasets
 
 class TestVTKDataSource(unittest.TestCase):
 

diff --git a/mayavi/tests/test_vtk_file_reader.py b/mayavi/tests/test_vtk_file_reader.py
@@ -11,7 +11,7 @@
 from mayavi.sources.vtk_file_reader import VTKFileReader
 
 # Local imports.
-from .common import get_example_data
+from mayavi.tests.common import get_example_data
 
 vtk_major_version = vtk.vtkVersion.GetVTKMajorVersion()
 vtk_minor_version = vtk.vtkVersion.GetVTKMinorVersion()

diff --git a/mayavi/tests/test_vtk_xml_reader.py b/mayavi/tests/test_vtk_xml_reader.py
@@ -19,7 +19,7 @@
 from mayavi.modules.scalar_cut_plane import ScalarCutPlane
 
 # Local imports.
-from .common import get_example_data
+from mayavi.tests.common import get_example_data
 
 class TestVTKXMLReader(unittest.TestCase):