X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Ftests%2Ftest_gameboard.py;fp=naja%2Ftests%2Ftest_gameboard.py;h=5dae954fafaae21a532b7a7d842b408f6d210101;hb=02ac35df9206ceaf9469ce5e77555b4809c1abcf;hp=bbc51284cfcb9847b6cfe8e8e194d482f126c802;hpb=0cef06f581d48315a59a19ac06ca1d2be14cbb4c;p=naja.git

diff --git a/naja/tests/test_gameboard.py b/naja/tests/test_gameboard.py
index bbc5128..5dae954 100644
--- a/naja/tests/test_gameboard.py
+++ b/naja/tests/test_gameboard.py
@@ -85,6 +85,45 @@ class TestGameBoard(TestCase):
 
         self.assert_state(state_1, state_2)
 
+    def generate_locations(self, override_dict=None):
+        locations_dict = dict(((x, y), '%s%s' % (x, y))
+                              for x in range(5) for y in range(5))
+        if override_dict:
+            locations_dict.update(override_dict)
+        return locations_dict
+
+    def test_shift_locations_north(self):
+        board = GameBoard.new_game([{'actions': []}])
+        board.board_locations = self.generate_locations()
+        board.shift_locations('NORTH')
+        self.assertEqual(board.board_locations, self.generate_locations({
+            (2, 0): '21', (2, 1): '23', (2, 3): '24', (2, 4): '20',
+        }))
+
+    def test_shift_locations_south(self):
+        board = GameBoard.new_game([{'actions': []}])
+        board.board_locations = self.generate_locations()
+        board.shift_locations('SOUTH')
+        self.assertEqual(board.board_locations, self.generate_locations({
+            (2, 0): '24', (2, 1): '20', (2, 3): '21', (2, 4): '23',
+        }))
+
+    def test_shift_locations_east(self):
+        board = GameBoard.new_game([{'actions': []}])
+        board.board_locations = self.generate_locations()
+        board.shift_locations('EAST')
+        self.assertEqual(board.board_locations, self.generate_locations({
+            (0, 2): '42', (1, 2): '02', (3, 2): '12', (4, 2): '32',
+        }))
+
+    def test_shift_locations_west(self):
+        board = GameBoard.new_game([{'actions': []}])
+        board.board_locations = self.generate_locations()
+        board.shift_locations('WEST')
+        self.assertEqual(board.board_locations, self.generate_locations({
+            (0, 2): '12', (1, 2): '32', (3, 2): '42', (4, 2): '02',
+        }))
+
 
 class TestLocationCard(TestCase):
     def test_generate_bitwise_operand(self):