From 27249422c17eaf0c223e34e7860fac4064927561 Mon Sep 17 00:00:00 2001 From: MRXLT Date: Fri, 10 Jul 2020 10:46:02 +0800 Subject: [PATCH] add assert --- python/examples/imagenet/test_image_reader.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/examples/imagenet/test_image_reader.py b/python/examples/imagenet/test_image_reader.py index 3f5beb2f..b3e1aac7 100644 --- a/python/examples/imagenet/test_image_reader.py +++ b/python/examples/imagenet/test_image_reader.py @@ -21,7 +21,7 @@ def test_String2Image(): img_str = f.read() seq = Sequential([String2Image()]) img = seq(img_str) - print(img.shape) + assert (img.shape == (563, 500, 3)) def test_Base64ToImage(): @@ -29,7 +29,7 @@ def test_Base64ToImage(): img_str = f.read() seq = Sequential([Base64ToImage()]) img = seq(base64.b64encode(img_str)) - print(img.shape) + assert (img.shape == (563, 500, 3)) if __name__ == "__main__": -- GitLab