From a704573e37defa141e3ecd548b14d2d2f85a458d Mon Sep 17 00:00:00 2001 From: Shanqing Cai Date: Sun, 15 Jan 2017 14:46:13 -0800 Subject: [PATCH] Fix a breakage in python3 test Change: 144584357 --- .../contrib/learn/python/learn/learn_io/graph_io_test.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tensorflow/contrib/learn/python/learn/learn_io/graph_io_test.py b/tensorflow/contrib/learn/python/learn/learn_io/graph_io_test.py index e9adb8cc048..90d58dec14e 100644 --- a/tensorflow/contrib/learn/python/learn/learn_io/graph_io_test.py +++ b/tensorflow/contrib/learn/python/learn/learn_io/graph_io_test.py @@ -727,13 +727,14 @@ class GraphIOTest(test.TestCase): self.assertAllEqual( [filename.encode("utf-8") + b":2", filename.encode("utf-8") + b":3"], out_keys) - self.assertAllEqual([lines[1], lines[2]], out_vals) + self.assertAllEqual([lines[1].encode("utf-8"), lines[2].encode("utf-8")], + out_vals) # Second batch will only have one filtered example as that's the only # remaining example that satisfies the filtering criterion. out_keys, out_vals = session.run((keys, inputs)) self.assertAllEqual([filename.encode("utf-8") + b":4"], out_keys) - self.assertAllEqual([lines[3]], out_vals) + self.assertAllEqual([lines[3].encode("utf-8")], out_vals) # Exhausted input. with self.assertRaises(errors.OutOfRangeError): -- GitLab