adding tests for agent directions

aimacode · kaivalyar · Mar 7, 2017 · Mar 7, 2017 · Mar 7, 2017 · Mar 7, 2017
commit ae0d9d16aa94b9252612af25c674dab88f3ee4ec
diff --git a/tests/test_agents.py b/tests/test_agents.py
@@ -0,0 +1,40 @@
+from agents import Direction
+
+def test_move_forward():
+    d = Direction("up")
+    l1 = d.move_forward((0,0))
+    assert l1 == (0,-1)
+    d = Direction(Direction.R)
+    l1 = d.move_forward((0,0))
+    assert l1 == (1,0)
+    d = Direction(Direction.D)
+    l1 = d.move_forward((0,0))
+    assert l1 == (0,1)
+    d = Direction("left")
+    l1 = d.move_forward((0,0))
+    assert l1 == (-1,0)
+    l2 = d.move_forward((1,0))
+    assert l2 == (0,0)
+
+def test_add():
+    d = Direction(Direction.U)
+    l1 = d + "right"
+    l2 = d + "left"
+    assert l1.direction == Direction.R
+    assert l2.direction == Direction.L
+    d = Direction("right")
+    l1 = d.__add__(Direction.L)
+    l2 = d.__add__(Direction.R)
+    assert l1.direction == "up"
+    assert l2.direction == "down"
+    d = Direction("down")
+    l1 = d.__add__("right")
+    l2 = d.__add__("left")
+    assert l1.direction == Direction.L
+    assert l2.direction == Direction.R
+    d = Direction(Direction.L)
+    l1 = d + Direction.R
+    l2 = d + Direction.L
+    assert l1.direction == Direction.U
+    #assert l2.direction == Direction.D currently gives an error - need to fix bug in agents.py
+