Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-09-01 19:49:03 +02:00
Code Issues Packages Projects Releases Wiki Activity

[to #48291018]fix: fix data releate case failed bug

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11925637
This commit is contained in:
mulin.lyh
2023-03-09 15:18:50 +08:00
committed by wenmeng.zwm
parent 46799325c8
commit 62f769877f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/pipelines/test_base.py
View File

@@ -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__':