From 46fc7a9530e9c8f6bf909de8df8c97e4b38a99a5 Mon Sep 17 00:00:00 2001 From: josephyearsley Date: Mon, 23 Jul 2018 23:06:48 +0100 Subject: Fixed Tests --- tensorflow/python/layers/core_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'tensorflow/python/layers') diff --git a/tensorflow/python/layers/core_test.py b/tensorflow/python/layers/core_test.py index 8ad0e8c4ba..22ed75dda7 100644 --- a/tensorflow/python/layers/core_test.py +++ b/tensorflow/python/layers/core_test.py @@ -491,7 +491,7 @@ class FlattenTest(test.TestCase): [0, 4, 1, 2, 3]) np_output_cf = sess.run(y, feed_dict={x: np_input_channels_first}) - self.assertEqual(np_output_cl, np_output_cf) + self.assertAllEqual(np_output_cl, np_output_cf) def testDataFormat4d(self): np_input_channels_last = np.arange(24, dtype='float32').reshape( @@ -508,7 +508,7 @@ class FlattenTest(test.TestCase): [0, 3, 1, 2]) np_output_cf = sess.run(y, feed_dict={x: np_input_channels_first}) - self.assertEqual(np_output_cl, np_output_cf) + self.assertAllEqual(np_output_cl, np_output_cf) def testFunctionalFlatten(self): x = array_ops.placeholder(shape=(None, 2, 3), dtype='float32') -- cgit v1.2.3