Merge pull request #294 from stan-dev/feature/services-exception

TST: update test checking gradient error exception message propogation

Author: ariddell

pystan/tests/test_rstan_stanfit.py

@@ -20,8 +20,7 @@ def test_init_zero_exception_inf_grad(self):
         }
         """
         sm = StanModel(model_code=code)
-        assertRaisesRegex = self.assertRaisesRegexp if PY2 else self.assertRaisesRegex
-        with assertRaisesRegex(RuntimeError, ''):
+        with self.assertRaises(RuntimeError):
             sm.sampling(init='0', iter=1, chains=1)
 
     def test_grad_log(self):

pystan/tests/test_user_inits.py

@@ -72,13 +72,9 @@ def test_user_init_unspecified(self):
         }
         """
         data = self.data
-        # NOTE: we are only specifying 'mu' and not 'sigma'
-        # This behavior works now.
-        try:
-            pystan.stan(model_code=model_code, iter=10, chains=1, seed=2,
-                        data=data, init=[dict(mu=4)], warmup=0)
-        except RuntimeError:
-            self.fail("pystan.stan() raised RuntimeError on partial init")
+        # NOTE: we are only specifying 'mu' and not 'sigma' (partial inits)
+        fit = pystan.stan(model_code=model_code, iter=10, chains=1, seed=2, data=data, init=[dict(mu=4)], warmup=0)
+        self.assertIsNotNone(fit)
 
 
 class TestUserInitsMatrix(unittest.TestCase):