MD UT - Add missing set and restore seed.

test_places365_content_check - delete visualize_dataset
This commit is contained in:
Cathy Wong 2021-10-26 11:14:50 -04:00
parent e17ba24394
commit e66aba9ef5
3 changed files with 12 additions and 2 deletions

View File

@ -71,8 +71,7 @@ def test_places365_content_check():
logger.info("Test Places365Dataset Op with content check")
sampler = ds.SequentialSampler(num_samples=4)
data1 = ds.Places365Dataset(dataset_dir=DATA_DIR, usage='val', small=True, decode=True, sampler=sampler)
images, labels = load_places365(DATA_DIR)
visualize_dataset(images, labels)
_, labels = load_places365(DATA_DIR)
num_iter = 0
# in this example, each dictionary has keys "image" and "label"
image_list, label_list = [], []

View File

@ -84,6 +84,8 @@ def test_random_posterize_op_fixed_point_c(plot=False, run_golden=True):
Test RandomPosterize in C transformations with fixed point
"""
logger.info("test_random_posterize_op_c")
original_seed = config_get_set_seed(55)
original_num_parallel_workers = config_get_set_num_parallel_workers(1)
# define map operations
transforms1 = [
@ -115,6 +117,10 @@ def test_random_posterize_op_fixed_point_c(plot=False, run_golden=True):
if plot:
visualize_list(image_original, image_posterize)
# Restore configuration
ds.config.set_seed(original_seed)
ds.config.set_num_parallel_workers(original_num_parallel_workers)
def test_random_posterize_default_c_md5(plot=False, run_golden=True):
"""

View File

@ -76,6 +76,8 @@ def test_random_solarize_mnist(plot=False, run_golden=True):
"""
Test RandomSolarize op with MNIST dataset (Grayscale images)
"""
original_seed = config_get_set_seed(0)
original_num_parallel_workers = config_get_set_num_parallel_workers(1)
mnist_1 = ds.MnistDataset(dataset_dir=MNIST_DATA_DIR, num_samples=2, shuffle=False)
mnist_2 = ds.MnistDataset(dataset_dir=MNIST_DATA_DIR, num_samples=2, shuffle=False)
@ -99,6 +101,9 @@ def test_random_solarize_mnist(plot=False, run_golden=True):
filename = "random_solarize_02_result.npz"
save_and_check_md5(mnist_2, filename, generate_golden=GENERATE_GOLDEN)
ds.config.set_seed(original_seed)
ds.config.set_num_parallel_workers(original_num_parallel_workers)
def test_random_solarize_errors():
"""