Browse Source

!1019 Cleanup dataset UT: use md5 npz in test_zip for images

Merge pull request !1019 from cathwong/ckw_dataset_ut_cleanup2
tags/v0.3.0-alpha
mindspore-ci-bot Gitee 5 years ago
parent
commit
37f787b657
5 changed files with 6 additions and 9 deletions
  1. BIN
      tests/ut/data/dataset/golden/zip_02_result.npz
  2. BIN
      tests/ut/data/dataset/golden/zip_03_result.npz
  3. BIN
      tests/ut/data/dataset/golden/zip_04_result.npz
  4. +2
    -5
      tests/ut/python/dataset/test_decode.py
  5. +4
    -4
      tests/ut/python/dataset/test_zip.py

BIN
tests/ut/data/dataset/golden/zip_02_result.npz View File


BIN
tests/ut/data/dataset/golden/zip_03_result.npz View File


BIN
tests/ut/data/dataset/golden/zip_04_result.npz View File


+ 2
- 5
tests/ut/python/dataset/test_decode.py View File

@@ -20,16 +20,13 @@ import mindspore.dataset.transforms.vision.c_transforms as vision
import numpy as np
import mindspore.dataset as ds
from mindspore import log as logger
from util import diff_mse


DATA_DIR = ["../data/dataset/test_tf_file_3_images/train-0000-of-0001.data"]
SCHEMA_DIR = "../data/dataset/test_tf_file_3_images/datasetSchema.json"


def diff_mse(in1, in2):
mse = (np.square(in1.astype(float) / 255 - in2.astype(float) / 255)).mean()
return mse * 100


def test_decode_op():
"""
Test Decode op


+ 4
- 4
tests/ut/python/dataset/test_zip.py View File

@@ -12,7 +12,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.
# ==============================================================================
from util import save_and_check_dict
from util import save_and_check_dict, save_and_check_md5
from mindspore import log as logger

import mindspore.dataset as ds
@@ -58,7 +58,7 @@ def test_zip_02():
dataz = ds.zip((data1, data2))
# Note: zipped dataset has 3 rows and 4 columns
filename = "zip_02_result.npz"
save_and_check_dict(dataz, filename, generate_golden=GENERATE_GOLDEN)
save_and_check_md5(dataz, filename, generate_golden=GENERATE_GOLDEN)


def test_zip_03():
@@ -72,7 +72,7 @@ def test_zip_03():
dataz = ds.zip((data1, data2))
# Note: zipped dataset has 3 rows and 7 columns
filename = "zip_03_result.npz"
save_and_check_dict(dataz, filename, generate_golden=GENERATE_GOLDEN)
save_and_check_md5(dataz, filename, generate_golden=GENERATE_GOLDEN)


def test_zip_04():
@@ -87,7 +87,7 @@ def test_zip_04():
dataz = ds.zip((data1, data2, data3))
# Note: zipped dataset has 3 rows and 9 columns
filename = "zip_04_result.npz"
save_and_check_dict(dataz, filename, generate_golden=GENERATE_GOLDEN)
save_and_check_md5(dataz, filename, generate_golden=GENERATE_GOLDEN)


def test_zip_05():


Loading…
Cancel
Save