tests for additional directions added

Author: fatroom

homework1.4/test.py

@@ -194,10 +194,10 @@ def test_dataset9(self):
                            [1.0, 0.0, 0.0],
                            [0.0, 0.0, 0.0]]
 
-        task.colors = [['green', 'red', 'green'],
-                       ['green', 'red', 'red'],
-                       ['green', 'red', 'green']]
-        task.measurements = ['red', 'green']
+        task.colors = [['green', 'green', 'green'],
+                       ['red', 'red', 'green'],
+                       ['green', 'green', 'green']]
+        task.measurements = ['red', 'red']
         task.motions = [[0, 0], [0, -1]]
         task.sensor_right = 1.0
         task.p_move = 1.0
@@ -214,10 +214,10 @@ def test_dataset10(self):
                            [0.0, 0.0, 0.0],
                            [0.0, 0.0, 0.0]]
 
-        task.colors = [['green', 'green', 'green'],
-                       ['red', 'red', 'red'],
-                       ['green', 'red', 'green']]
-        task.measurements = ['red', 'green']
+        task.colors = [['green', 'red', 'green'],
+                       ['green', 'red', 'green'],
+                       ['green', 'green', 'green']]
+        task.measurements = ['red', 'red']
         task.motions = [[0, 0], [-1, 0]]
         task.sensor_right = 1.0
         task.p_move = 1.0
@@ -227,5 +227,6 @@ def test_dataset10(self):
 
         # ASSERT
         self.assertArrayAlmostEquals(expected_result, actual_result, 4)
+
 if __name__ == '__main__':
     unittest.main()