#5 better test

ingambe · Mar 1, 2021 · db67e53 · db67e53
1 parent 518125c
commit db67e53
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/tests/flexible/test_fexible_random.py b/tests/flexible/test_fexible_random.py
@@ -8,15 +8,18 @@ def test_random(self):
         env = gym.make('JSSEnv:flexible-jss-v1', env_config={'instance_path': '../../JSSEnv/envs/instances/flexible/mt10c1.fjs'})
 
         average = 0
-        for _ in range(1):
+        for _ in range(100):
             state = env.reset()
             self.assertEqual(env.current_time_step, 0)
+            self.assertNotEqual(env.todo_time_step_job.all(), env.nb_op_job.all())
+            self.assertEqual(env.todo_time_step_job.all(), 0)
             done = False
             while not done:
                 legal_actions = env.get_legal_actions()
-                first_legal_action = np.where(legal_actions)[0][0]
-                #print(first_legal_action)
-                state, reward, done, _ = env.step(first_legal_action)
+                actions = np.random.choice(len(legal_actions), 1, p=(legal_actions / legal_actions.sum()))[0]
+                state, reward, done, _ = env.step(actions)
+            print(env.current_time_step)
+            self.assertEqual(env.todo_time_step_job.all(), env.nb_op_job.all())
         '''
         average = 0
         for _ in range(100):