Kanvi/Add asserts in some python tests (#398)

tensorflow · Dec 20, 2019 · 367d3db · 367d3db
1 parent b267ba1
commit 367d3db
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 12 deletions.
diff --git a/test/python/test_sigmoid.py b/test/python/test_sigmoid.py
@@ -52,4 +52,5 @@ def test_sigmoid(self):
                                             y: y_np,
                                             z: z_np
                                         })
-        np.allclose(self.with_ngraph(sess_fn), self.without_ngraph(sess_fn))
+        assert np.allclose(
+            self.with_ngraph(sess_fn), self.without_ngraph(sess_fn))
diff --git a/test/python/test_sign.py b/test/python/test_sign.py
@@ -35,14 +35,16 @@ def test_sign_1d(self, test_input, expected):
         val = tf.placeholder(tf.float32, shape=(1,))
         out = tf.sign(val)
         sess_fn = lambda sess: sess.run((out,), feed_dict={val: (test_input,)})
-        np.allclose(self.with_ngraph(sess_fn), self.without_ngraph(sess_fn))
-        np.allclose(self.with_ngraph(sess_fn), expected)
+        assert np.allclose(
+            self.with_ngraph(sess_fn), self.without_ngraph(sess_fn))
+        assert np.allclose(self.with_ngraph(sess_fn), expected)
 
     def test_sign_2d(self):
         test_input = ((1.5, -2.5, -3.5), (-4.5, 5.5, 0))
         expected = ((1, -1, -1), (-1, 1, 0))
         val = tf.placeholder(tf.float32, shape=(2, 3))
         out = tf.sign(val)
         sess_fn = lambda sess: sess.run((out,), feed_dict={val: test_input})
-        np.allclose(self.with_ngraph(sess_fn), self.without_ngraph(sess_fn))
-        np.allclose(self.with_ngraph(sess_fn), expected)
+        assert np.allclose(
+            self.with_ngraph(sess_fn), self.without_ngraph(sess_fn))
+        assert np.allclose(self.with_ngraph(sess_fn), expected)
diff --git a/test/python/test_softmax.py b/test/python/test_softmax.py
@@ -43,8 +43,9 @@ def test_softmax_2d(self):
         expected = a_np
         a = tf.nn.softmax(x)
         sess_fn = lambda sess: sess.run((a), feed_dict={x: x_np})
-        np.allclose(self.with_ngraph(sess_fn), self.without_ngraph(sess_fn))
-        np.allclose(self.with_ngraph(sess_fn), expected)
+        assert np.allclose(
+            self.with_ngraph(sess_fn), self.without_ngraph(sess_fn))
+        assert np.allclose(self.with_ngraph(sess_fn), expected)
 
     def test_softmax_3d(self):
         x = tf.placeholder(tf.float32, shape=(2, 3, 2))
@@ -59,8 +60,9 @@ def test_softmax_3d(self):
         expected = a_np
         a = tf.nn.softmax(x)
         sess_fn = lambda sess: sess.run((a), feed_dict={x: x_np})
-        np.allclose(self.with_ngraph(sess_fn), self.without_ngraph(sess_fn))
-        np.allclose(self.with_ngraph(sess_fn), expected)
+        assert np.allclose(
+            self.with_ngraph(sess_fn), self.without_ngraph(sess_fn))
+        assert np.allclose(self.with_ngraph(sess_fn), expected)
 
     def test_softmax_4d(self):
         x = tf.placeholder(tf.float32, shape=(2, 3, 2, 4))
@@ -75,5 +77,6 @@ def test_softmax_4d(self):
         expected = a_np
         a = tf.nn.softmax(x)
         sess_fn = lambda sess: sess.run((a), feed_dict={x: x_np})
-        np.allclose(self.with_ngraph(sess_fn), self.without_ngraph(sess_fn))
-        np.allclose(self.with_ngraph(sess_fn), expected)
+        assert np.allclose(
+            self.with_ngraph(sess_fn), self.without_ngraph(sess_fn))
+        assert np.allclose(self.with_ngraph(sess_fn), expected)
diff --git a/test/python/test_stack.py b/test/python/test_stack.py
@@ -50,4 +50,5 @@ def test_stack(self, shapes, axis):
         a = tf.stack(placeholders, axis)
         sess_fn = lambda sess: sess.run(
             [a], feed_dict={p: v for p, v in zip(placeholders, values)})
-        np.allclose(self.with_ngraph(sess_fn), self.without_ngraph(sess_fn))
+        assert np.allclose(
+            self.with_ngraph(sess_fn), self.without_ngraph(sess_fn))