p12346870
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Replace print with logger.info

This commit is contained in:
Jesse Lee 2020-04-29 12:44:04 -04:00
parent b1ea933345
commit 5236d0c3c0
1 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
tests/ut/python/dataset/test_random_dataset.py
View File

@ -14,11 +14,12 @@
# ============================================================================== # ==============================================================================
import mindspore.common.dtype as mstype import mindspore.common.dtype as mstype
import mindspore.dataset as ds import mindspore.dataset as ds
from mindspore import log as logger
from pathlib import Path from pathlib import Path
# just a basic test with parallel random data op # just a basic test with parallel random data op
def test_randomdataset_basic1(): def test_randomdataset_basic1():
print("Test randomdataset basic") logger.info("Test randomdataset basic")
schema = ds.Schema() schema = ds.Schema()
schema.add_column('image', de_type=mstype.uint8, shape=[2]) schema.add_column('image', de_type=mstype.uint8, shape=[2])
@ -31,16 +32,16 @@ def test_randomdataset_basic1():
num_iter = 0 num_iter = 0
for data in ds1.create_dict_iterator(): # each data is a dictionary for data in ds1.create_dict_iterator(): # each data is a dictionary
# in this example, each dictionary has keys "image" and "label" # in this example, each dictionary has keys "image" and "label"
print("{} image: {}".format(num_iter, data["image"])) logger.info("{} image: {}".format(num_iter, data["image"]))
print("{} label: {}".format(num_iter, data["label"])) logger.info("{} label: {}".format(num_iter, data["label"]))
num_iter += 1 num_iter += 1
print("Number of data in ds1: ", num_iter) logger.info("Number of data in ds1: ", num_iter)
assert(num_iter == 200) assert(num_iter == 200)
# Another simple test # Another simple test
def test_randomdataset_basic2(): def test_randomdataset_basic2():
print("Test randomdataset basic 2") logger.info("Test randomdataset basic 2")
schema = ds.Schema() schema = ds.Schema()
schema.add_column('image', de_type=mstype.uint8, shape=[640,480,3]) # 921600 bytes (a bit less than 1 MB per image) schema.add_column('image', de_type=mstype.uint8, shape=[640,480,3]) # 921600 bytes (a bit less than 1 MB per image)
@ -55,16 +56,16 @@ def test_randomdataset_basic2():
num_iter = 0 num_iter = 0
for data in ds1.create_dict_iterator(): # each data is a dictionary for data in ds1.create_dict_iterator(): # each data is a dictionary
# in this example, each dictionary has keys "image" and "label" # in this example, each dictionary has keys "image" and "label"
#print(data["image"]) #logger.info(data["image"])
print("printing the label: {}".format(data["label"])) logger.info("printing the label: {}".format(data["label"]))
num_iter += 1 num_iter += 1
print("Number of data in ds1: ", num_iter) logger.info("Number of data in ds1: ", num_iter)
assert(num_iter == 40) assert(num_iter == 40)
if __name__ == '__main__': if __name__ == '__main__':
test_randomdataset_basic1() test_randomdataset_basic1()
test_randomdataset_basic2() test_randomdataset_basic2()
print('test_randomdataset_basic Ended.\n') logger.info('test_randomdataset_basic Ended.\n')