From 8c88b39da1328fe9054209891003ea7916adc6bc Mon Sep 17 00:00:00 2001
From: qianlong <qianlong3@huawei.com>
Date: Mon, 30 Mar 2020 20:16:45 +0800
Subject: [PATCH] Optimize the execution time of test case test_rgb_hsv.py

---
 tests/ut/python/dataset/test_rgb_hsv.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/ut/python/dataset/test_rgb_hsv.py b/tests/ut/python/dataset/test_rgb_hsv.py
index e887acb01d..14ab9ddc2d 100644
--- a/tests/ut/python/dataset/test_rgb_hsv.py
+++ b/tests/ut/python/dataset/test_rgb_hsv.py
@@ -133,6 +133,7 @@ def test_rgb_hsv_pipeline():
     # First dataset
     transforms1 = [
         vision.Decode(),
+        vision.Resize([64, 64]),
         vision.ToTensor()
     ]
     transforms1 = vision.ComposeOp(transforms1)
@@ -142,6 +143,7 @@ def test_rgb_hsv_pipeline():
     # Second dataset
     transforms2 = [
         vision.Decode(),
+        vision.Resize([64, 64]),
         vision.ToTensor(),
         vision.RgbToHsv(),
         vision.HsvToRgb()