experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Eager execution coverage for decode_image_op_test.py. Removed run_deprecated_v1 decorators.

Browse Source 
PiperOrigin-RevId: 327043628
Change-Id: Ice7843c3fe9991d3cbad0e0608231c180213e5d4
This commit is contained in:
Hye Soo Yang 2020-08-17 10:07:18 -07:00 committed by TensorFlower Gardener
parent fa9564ee72
commit 91bcb88f53
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/python/kernel_tests/decode_image_op_test.py
View File

@ -45,7 +45,6 @@ class DecodeImageOpTest(test.TestCase):
self.assertEqual(len(bmp0), 4194)
self.assertAllEqual(image0, image1)
@test_util.run_deprecated_v1
def testGif(self):
# Read some real GIFs
path = os.path.join(prefix_path, "gif", "testdata", "scan.gif")
@ -76,11 +75,10 @@ class DecodeImageOpTest(test.TestCase):
self.assertAllClose(frame, gt)
bad_channels = image_ops.decode_image(gif0, channels=1)
with self.assertRaises(errors_impl.InvalidArgumentError):
bad_channels = image_ops.decode_image(gif0, channels=1)
self.evaluate(bad_channels)
@test_util.run_deprecated_v1
def testJpeg(self):
# Read a real jpeg and verify shape
path = os.path.join(prefix_path, "jpeg", "testdata", "jpeg_merge_test1.jpg")
@ -93,8 +91,8 @@ class DecodeImageOpTest(test.TestCase):
self.assertEqual(image0.shape, (256, 128, 3))
self.assertAllEqual(image0, image1)
bad_channels = image_ops.decode_image(jpeg0, channels=4)
with self.assertRaises(errors_impl.InvalidArgumentError):
bad_channels = image_ops.decode_image(jpeg0, channels=4)
self.evaluate(bad_channels)
def testPng(self):