diff --git a/MetaAugment/autoaugment_learners/parse_ds_cn_arch.py b/temp_util/parse_ds_cn_arch.py
similarity index 100%
rename from MetaAugment/autoaugment_learners/parse_ds_cn_arch.py
rename to temp_util/parse_ds_cn_arch.py
diff --git a/wapp_util.py b/temp_util/wapp_util.py
similarity index 100%
rename from wapp_util.py
rename to temp_util/wapp_util.py
diff --git a/test/MetaAugment/test_ucb_learner.py b/test/MetaAugment/test_ucb_learner.py
new file mode 100644
index 0000000000000000000000000000000000000000..514d78307eb553afd16521309e4273127f3fa40e
--- /dev/null
+++ b/test/MetaAugment/test_ucb_learner.py
@@ -0,0 +1,25 @@
+import MetaAugment.autoaugment_learners as aal
+import MetaAugment.child_networks as cn
+import torch
+import torchvision
+import torchvision.datasets as datasets
+
+import random
+
+
+def test_ucb_learner():
+    policies = UCB1_JC.generate_policies(num_policies, num_sub_policies)
+        q_values, best_q_values = UCB1_JC.run_UCB1(
+                                                policies,
+                                                batch_size, 
+                                                learning_rate, 
+                                                ds, 
+                                                toy_size, 
+                                                max_epochs, 
+                                                early_stop_num, 
+                                                iterations, 
+                                                IsLeNet, 
+                                                ds_name
+                                                )     
+        best_q_values = np.array(best_q_values)
+    pass
\ No newline at end of file