Fix Errors

pgmpy · Jan 21, 2017 · c4e013e · c4e013e
1 parent c77aa6d
commit c4e013e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pgmpy/tests/test_factors/test_continuous/test_ContinuousFactor.py b/pgmpy/tests/test_factors/test_continuous/test_ContinuousFactor.py
@@ -379,9 +379,9 @@ def test_discretize(self):
         results3 = univariate_chi.discretize(RoundingDiscretizer, low=0, high=1, cardinality=3)
         results3 = [round(i, 6) for i in results3]
 
-        self.assertAlmostEqual(results1, [0.154375, 0.085531, 0.0, -0.085531], places=4)
-        self.assertAlmostEqual(results2, [-2.120911, -0.115055, 0.198042, 0.033288, 0.004727], places=4)
-        self.assertAlmostEqual(results3, [0.038335, 0.059015, 0.039992], places=4)
+        np.testing.assert_almost_equal(results1, [0.154375, 0.085531, 0.0, -0.085531],decimal=4)
+        np.testing.assert_almost_equal(results2, [-2.120911, -0.115055, 0.198042, 0.033288, 0.004727], decimal=4)
+        np.testing.assert_almost_equal(results3, [0.038335, 0.059015, 0.039992], decimal=4)
 
     def test_copy(self):
         copy1 = self.phi1.copy()