update /test

test/MetaAugment/test_aa_learner.py

@@ -26,8 +26,8 @@ def test_translate_operation_tensor():
         softmax = torch.nn.Softmax(dim=0)
 
         fun_num = random.randint(1, 14)
-        p_bins = random.randint(1, 15)
-        m_bins = random.randint(1, 15)
+        p_bins = random.randint(2, 15)
+        m_bins = random.randint(2, 15)
 
         agent = aal.aa_learner(
                 sp_num=5,

test/MetaAugment/test_gru_learner.py

@@ -15,8 +15,8 @@ def test_generate_new_policy():
     for _ in range(40):
         sp_num = random.randint(1,20)
         fun_num = random.randint(1, 14)
-        p_bins = random.randint(1, 15)
-        m_bins = random.randint(1, 15)
+        p_bins = random.randint(2, 15)
+        m_bins = random.randint(2, 15)
 
         agent = aal.gru_learner(
             sp_num=sp_num,
@@ -24,7 +24,7 @@ def test_generate_new_policy():
             p_bins=p_bins,
             m_bins=m_bins
             )
-        for _ in range(10):
+        for _ in range(4):
             new_policy = agent.generate_new_policy()
             assert isinstance(new_policy[0], list), new_policy
 

test/MetaAugment/test_randomsearch_learner.py

@@ -17,8 +17,8 @@ def test_generate_new_policy():
         for _ in range(40):
             sp_num = random.randint(1,20)
             fun_num = random.randint(1, 14)
-            p_bins = random.randint(1, 15)
-            m_bins = random.randint(1, 15)
+            p_bins = random.randint(2, 15)
+            m_bins = random.randint(2, 15)
 
             agent = aal.randomsearch_learner(
                 sp_num=sp_num,
@@ -27,7 +27,7 @@ def test_generate_new_policy():
                 m_bins=m_bins,
                 discrete_p_m=discrete_p_m
                 )
-            for _ in range(10):
+            for _ in range(4):
                 new_policy = agent.generate_new_policy()
                 assert isinstance(new_policy, list), new_policy
     

test/README.md

-This directory contatins all unit tests and integration tests for this project,
-including for the library and the web-app.
+This directory contains all unit tests and integration tests for this project.
+
+The tests for the library can be seen in /test/MetaAugment.
+The tests for the web-app can be seen in ????
 
 Use command `pytest` in main directory to run all tests in this directory.
\ No newline at end of file