Merge pull request #583 from zploskey/testfixes

Fix some tests/examples for python 3 or if missing statsmodels

Author: twiecki

pymc/examples/glm_linear.py

@@ -1,4 +1,7 @@
+from __future__ import print_function
+
 import numpy as np
+import sys
 
 try:
     import statsmodels.api as sm

pymc/examples/glm_robust.py

@@ -1,8 +1,12 @@
+from __future__ import print_function
+
 import numpy as np
+import sys
 
 try:
     import statsmodels.api as sm
 except ImportError:
+    print("Example requires statsmodels")
     sys.exit(0)
 
 from pymc import *

pymc/tests/test_glm.py

@@ -6,7 +6,7 @@
 try:
     import statsmodels.api as sm
 except ImportError:
-    sys.exit(0)
+    raise SkipTest("Test requires statsmodels.")
 
 from pymc.examples import glm_linear, glm_robust