From 37b53f2337e2598750c8d2e38cb3d88619b961cf Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Sat, 10 Aug 2019 16:38:28 -0400
Subject: [PATCH] typos and minor updates

---
 tests/test_flatland_malfunction.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/test_flatland_malfunction.py b/tests/test_flatland_malfunction.py
index 4122d56f..2e82d212 100644
--- a/tests/test_flatland_malfunction.py
+++ b/tests/test_flatland_malfunction.py
@@ -94,16 +94,16 @@ def test_malfunction_process():
         obs, all_rewards, done, _ = env.step(actions)
 
         if agent_malfunctioning:
+            # Check that agent is not moving while malfunctioning
             assert agent_old_position == env.agents[0].position
 
         agent_old_position = env.agents[0].position
         total_down_time += env.agents[0].malfunction_data['malfunction']
 
-
-    # Check that the agents breaks twice
+    # Check that the appropriate number of malfunctions is achieved
     assert env.agents[0].malfunction_data['nr_malfunctions'] == 5
 
-    # Check that 11 stops where performed
+    # Check that 20 stops where performed
     assert agent_halts == 20
 
     # Check that malfunctioning data was standing around
-- 
GitLab