Merge pull request #279 from bashtage/test-pyargs

BLD: Use pyargs to test

arch/__init__.py

@@ -1,5 +1,6 @@
 from arch._version import get_versions
 from arch.univariate.mean import arch_model
+from arch.utility import test
 
 __version__ = get_versions()['version']
 del get_versions
@@ -10,4 +11,4 @@ def doc():
     webbrowser.open('http://arch.readthedocs.org/en/latest/')
 
 
-__all__ = ['arch_model', '__version__', 'doc']
+__all__ = ['arch_model', '__version__', 'doc', 'test']

arch/utility/__init__.py

@@ -1,3 +1,24 @@
+import os
+import sys
+
 from arch.utility.cov import cov_nw
 
-__all__ = ['cov_nw']
+PKG = os.path.dirname(os.path.dirname(__file__))
+
+
+def test(extra_args=None):
+    try:
+        import pytest
+    except ImportError:
+        raise ImportError("Need pytest to run tests")
+    cmd = []
+    if extra_args:
+        if not isinstance(extra_args, list):
+            extra_args = [extra_args]
+        cmd = extra_args
+    cmd += [PKG]
+    print("running: pytest {}".format(' '.join(cmd)))
+    sys.exit(pytest.main(cmd))
+
+
+__all__ = ['cov_nw', 'test']