diff --git a/lcviz/__init__.py b/lcviz/__init__.py
index bb838887..3f9e1316 100644
--- a/lcviz/__init__.py
+++ b/lcviz/__init__.py
@@ -1,5 +1,8 @@
 # Licensed under a 3-clause BSD style license - see LICENSE.rst
 
+import os
+from astropy.tests.runner import TestRunner
+
 try:
     from .version import version as __version__
 except ImportError:
@@ -15,3 +18,6 @@
 from .tools import *  # noqa
 from .viewers import *  # noqa
 from .helper import *  # noqa
+
+# Create the test function for self test
+test = TestRunner.make_test_runner_in(os.path.dirname(__file__))