From e4d6cb9744d94d9a84369f4b8afda6ce89f1d1d3 Mon Sep 17 00:00:00 2001
From: "Joshua V. Dillon" <jvdillon@google.com>
Date: Thu, 20 Jul 2017 16:19:22 -0700
Subject: [PATCH] Change contrib/bayesflow/custom_grad tests to assertAllClose.
PiperOrigin-RevId: 162681035
---
.../bayesflow/python/kernel_tests/custom_grad_test.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/tensorflow/contrib/bayesflow/python/kernel_tests/custom_grad_test.py b/tensorflow/contrib/bayesflow/python/kernel_tests/custom_grad_test.py
index 5f8f7f692cf..a95df31ac1f 100644
--- a/tensorflow/contrib/bayesflow/python/kernel_tests/custom_grad_test.py
+++ b/tensorflow/contrib/bayesflow/python/kernel_tests/custom_grad_test.py
@@ -48,8 +48,8 @@ class CustomGradientTest(test.TestCase):
gx = gradients_impl.gradients(fx, x)[0]
[fx_, gx_] = sess.run([fx, gx])
- self.assertAllEqual(f(x_), fx_)
- self.assertAllEqual(g(x_), gx_)
+ self.assertAllClose(f(x_), fx_)
+ self.assertAllClose(g(x_), gx_)
def test_works_correctly_both_f_g_zero(self):
with self.test_session() as sess:
@@ -62,8 +62,8 @@ class CustomGradientTest(test.TestCase):
gx = gradients_impl.gradients(fx, x)[0]
[fx_, gx_] = sess.run([fx, gx])
- self.assertAllEqual(f(x_), fx_)
- self.assertAllEqual(g(x_), gx_)
+ self.assertAllClose(f(x_), fx_)
+ self.assertAllClose(g(x_), gx_)
def test_works_correctly_vector_of_vars(self):
with self.test_session() as sess: