From 62f769877fb272513b2997bf7fd30fdbbab7b150 Mon Sep 17 00:00:00 2001 From: "mulin.lyh" Date: Thu, 9 Mar 2023 15:18:50 +0800 Subject: [PATCH] [to #48291018]fix: fix data releate case failed bug Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11925637 --- tests/pipelines/test_base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/pipelines/test_base.py b/tests/pipelines/test_base.py index 75aa8afc..32bf8877 100644 --- a/tests/pipelines/test_base.py +++ b/tests/pipelines/test_base.py @@ -179,13 +179,13 @@ class CustomPipelineTest(unittest.TestCase): img_url = 'data/test/images/dogs.jpg' output = pipe(img_url) self.assertEqual(output['filename'], img_url) - self.assertEqual(output[OutputKeys.OUTPUT_IMG].shape, (318, 512, 3)) + self.assertEqual(output[OutputKeys.OUTPUT_IMG].shape, (598, 600, 3)) outputs = pipe([img_url for i in range(4)]) self.assertEqual(len(outputs), 4) for out in outputs: self.assertEqual(out['filename'], img_url) - self.assertEqual(out[OutputKeys.OUTPUT_IMG].shape, (318, 512, 3)) + self.assertEqual(out[OutputKeys.OUTPUT_IMG].shape, (598, 600, 3)) if __name__ == '__main__':