test random

Author: aloctavodia

pymc3/tests/test_bart.py

@@ -1,6 +1,7 @@
 import numpy as np
 
 from numpy.random import RandomState
+from numpy.testing import assert_almost_equal
 
 import pymc3 as pm
 
@@ -61,6 +62,6 @@ def test_bart_random():
     rng = RandomState(12345)
     pred_first = mu.owner.op.rng_fn(rng, X_new=X[:10])
 
-    assert np.all(pred_first == pred_all[0, :10])
+    assert_almost_equal(pred_first, pred_all[0, :10], decimal=4)
     assert pred_all.shape == (2, 50)
     assert pred_first.shape == (10,)