Fixed float32 precision errors

Author: lucianopaz

pymc3/tests/test_distributions_timeseries.py

@@ -6,6 +6,7 @@
 
 import numpy as np
 import pytest
+from .helpers import select_by_precision
 
 pytestmark = pytest.mark.usefixtures('seeded_test')
 
@@ -92,7 +93,8 @@ def test_GARCH11():
         z = Normal('z', mu=0, sigma=vol, shape=data.shape)
     garch_like = t['y'].logp({'z':data, 'y': data})
     reg_like = t['z'].logp({'z':data, 'y': data})
-    np.testing.assert_allclose(garch_like, reg_like)
+    decimal = select_by_precision(float64=7, float32=4)
+    np.testing.assert_allclose(garch_like, reg_like, 10**(-decimal))
 
 
 

pymc3/tests/test_minibatches.py

@@ -232,7 +232,14 @@ def test_multidim_scaling(self):
         with pm.Model() as model5:
             Normal('n', observed=[[1]], total_size=[2, Ellipsis, 2])
             p5 = theano.function([], model5.logpt)
-        assert p0() == p1() == p2() == p3() == p4() == p5()
+        _p0 = p0()
+        assert (
+            np.allclose(_p0, p1()) and
+            np.allclose(_p0, p2()) and
+            np.allclose(_p0, p3()) and
+            np.allclose(_p0, p4()) and
+            np.allclose(_p0, p5())
+        )
 
     def test_common_errors(self):
         with pm.Model():