From e66aba9ef53bdfd2f31a777f6c1db5f9c6c6e763 Mon Sep 17 00:00:00 2001
From: Cathy Wong <cathy.wong@huawei.com>
Date: Tue, 26 Oct 2021 11:14:50 -0400
Subject: [PATCH] MD UT - Add missing set and restore seed.
 test_places365_content_check - delete visualize_dataset

---
 tests/ut/python/dataset/test_datasets_places365.py | 3 +--
 tests/ut/python/dataset/test_random_posterize.py   | 6 ++++++
 tests/ut/python/dataset/test_random_solarize_op.py | 5 +++++
 3 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/tests/ut/python/dataset/test_datasets_places365.py b/tests/ut/python/dataset/test_datasets_places365.py
index 934165b5d2c..eff3d838bca 100644
--- a/tests/ut/python/dataset/test_datasets_places365.py
+++ b/tests/ut/python/dataset/test_datasets_places365.py
@@ -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 = [], []
diff --git a/tests/ut/python/dataset/test_random_posterize.py b/tests/ut/python/dataset/test_random_posterize.py
index 34a8bd9c42c..f07e70e3c2e 100644
--- a/tests/ut/python/dataset/test_random_posterize.py
+++ b/tests/ut/python/dataset/test_random_posterize.py
@@ -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):
     """
diff --git a/tests/ut/python/dataset/test_random_solarize_op.py b/tests/ut/python/dataset/test_random_solarize_op.py
index dfe72ba598a..6389e1c390a 100644
--- a/tests/ut/python/dataset/test_random_solarize_op.py
+++ b/tests/ut/python/dataset/test_random_solarize_op.py
@@ -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():
     """