Can test if a rolling stone reaches its goal.

tests/rollingstone_tests.py

@@ -0,0 +1,27 @@
+
+import unittest
+from robotgame.logic.rollingstone import *
+from robotgame.logic.direction import *
+
+
+playfield_example_succeed = [
+    [Start(Down), None,    None,        None  ],
+    [None,        None,    Stone(),     None  ],
+    [Turn(Right), None,    Turn(Down),  None  ],
+    [None,        Stone(), Turn(Right), Goal()],
+    ]
+
+playfield_example_fail = [
+    [Start(Down), None,    None,        None  ],
+    [None,        None,    Stone(),     None  ],
+    [Turn(Right), Stone(), Turn(Down),  None  ],
+    [None,        None,    Turn(Right), Goal()],
+    ]
+
+class RollingStoneTest(unittest.TestCase):
+    def test_playfield(self):
+        self.assertTrue(reaches_goal(playfield_example_succeed, 100))
+        self.assertFalse(reaches_goal(playfield_example_fail, 100))
+
+if __name__ == '__main__':
+    unittest.main()