X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Ftests%2Ftest_gameboard.py;h=e3aa32e03be48a27e0da319219fa880e25491c77;hb=75d1f94ef500df4bea81d8988757d37221a310dd;hp=6231ac7750f96941d1bd40529890465ceb740772;hpb=266021313a1501c86c11c803851c2b4bb50a8c00;p=naja.git

diff --git a/naja/tests/test_gameboard.py b/naja/tests/test_gameboard.py
index 6231ac7..e3aa32e 100644
--- a/naja/tests/test_gameboard.py
+++ b/naja/tests/test_gameboard.py
@@ -21,10 +21,15 @@ class TestGameBoard(TestCase):
         self.assertEqual(state1, state2)
 
     def test_export_new_board(self):
-        board = GameBoard.new_game([{'actions': [{
-            'action_class': 'LoseHealthOrMSB',
-            'required_bits': [],
-        }]}])
+        board = GameBoard.new_game([{'actions': [
+            {
+                'action_class': 'LoseHealthOrMSB',
+                'required_bits': [],
+            }, {
+                'action_class': 'GainHealth',
+                'required_bits': [BITS.RED],
+            },
+        ]}])
         exported_state = board.export()
         board_locations = exported_state.pop('board_locations')
         self.assertEqual(exported_state, {
@@ -32,24 +37,36 @@ class TestGameBoard(TestCase):
             'health': 4,
             'wins_required': 4,
             'wins': 0,
-            'locations': [{'actions': [{
-                'action_class': 'LoseHealthOrMSB',
-                'required_bits': [],
-            }]}],
+            'locations': [{'actions': [
+                {
+                    'action_class': 'LoseHealthOrMSB',
+                    'required_bits': [],
+                }, {
+                    'action_class': 'GainHealth',
+                    'required_bits': [BITS.RED],
+                },
+            ]}],
             'player': board.player.export(),
         })
-        self.assertEqual(
-            set(board_locations.keys()),
-            set((x, y) for x in range(5) for y in range(5)))
-        for location_state in board_locations.values():
+        positions = []
+        for position, location_state in board_locations:
+            positions.append(position)
             self.assertEqual(
                 sorted(location_state.keys()), ['actions', 'bitwise_operand'])
-            self.assertEqual(location_state['actions'], [{
-                'action_class': 'LoseHealthOrMSB',
-                'required_bits': [],
-                'data': {},
-            }])
+            self.assertEqual(location_state['actions'], [
+                {
+                    'action_class': 'LoseHealthOrMSB',
+                    'required_bits': [],
+                    'data': {},
+                }, {
+                    'action_class': 'GainHealth',
+                    'required_bits': [BITS.RED],
+                    'data': {},
+                },
+            ])
             self.assertTrue(2 <= len(location_state['bitwise_operand']) <= 3)
+        self.assertEqual(
+            positions, sorted((x, y) for x in range(5) for y in range(5)))
 
     def test_lose_health(self):
         board = GameBoard.new_game([{'actions': []}])
@@ -124,21 +141,66 @@ class TestGameBoard(TestCase):
             (0, 2): '12', (1, 2): '32', (3, 2): '42', (4, 2): '02',
         }))
 
-    def test_allow_chess_move_knight(self):
+    def test_rotate_locations_clockwise(self):
         board = GameBoard.new_game([{'actions': []}])
         board.board_locations = self.generate_locations()
+        board.rotate_locations('CLOCKWISE')
+        self.assertEqual(board.board_locations, self.generate_locations({
+            (1, 1): '21', (2, 1): '31', (3, 1): '32',
+            (1, 2): '11',               (3, 2): '33',
+            (1, 3): '12', (2, 3): '13', (3, 3): '23',
+        }))
+
+    def test_rotate_locations_clockwise_top(self):
+        board = GameBoard.new_game([{'actions': []}], initial_pos=(2, 0))
+        board.board_locations = self.generate_locations()
+        board.rotate_locations('CLOCKWISE')
+        self.assertEqual(board.board_locations, self.generate_locations({
+            (1, 0): '30',               (3, 0): '31',
+            (1, 1): '10', (2, 1): '11', (3, 1): '21',
+        }))
+
+    def test_rotate_locations_clockwise_right(self):
+        board = GameBoard.new_game([{'actions': []}], initial_pos=(0, 2))
+        board.board_locations = self.generate_locations()
+        board.rotate_locations('CLOCKWISE')
+        self.assertEqual(board.board_locations, self.generate_locations({
+            (0, 1): '11', (1, 1): '12',
+                          (1, 2): '13',
+            (0, 3): '01', (1, 3): '03',
+        }))
+
+    def test_rotate_locations_clockwise_corner(self):
+        board = GameBoard.new_game([{'actions': []}], initial_pos=(0, 4))
+        board.board_locations = self.generate_locations()
+        board.rotate_locations('CLOCKWISE')
+        self.assertEqual(board.board_locations, self.generate_locations({
+            (0, 3): '13', (1, 3): '14',
+                          (1, 4): '03',
+        }))
+
+    def test_rotate_locations_anticlockwise(self):
+        board = GameBoard.new_game([{'actions': []}])
+        board.board_locations = self.generate_locations()
+        board.rotate_locations('ANTICLOCKWISE')
+        self.assertEqual(board.board_locations, self.generate_locations({
+            (1, 1): '12', (2, 1): '11', (3, 1): '21',
+            (1, 2): '13',               (3, 2): '31',
+            (1, 3): '23', (2, 3): '33', (3, 3): '32',
+        }))
+
+    def test_allow_chess_move_knight(self):
+        board = GameBoard.new_game([{'actions': []}])
         board.allow_chess_move(MOVES.KNIGHT)
         self.assertEqual(board.player.movement_mode, MOVES.KNIGHT)
 
     def test_allow_chess_move_bishop(self):
         board = GameBoard.new_game([{'actions': []}])
-        board.board_locations = self.generate_locations()
         board.allow_chess_move(MOVES.BISHOP)
         self.assertEqual(board.player.movement_mode, MOVES.BISHOP)
 
     def test_allow_chess_move_castle(self):
         board = GameBoard.new_game([{'actions': []}])
-        board.board_locations = self.generate_locations()
         board.allow_chess_move(MOVES.CASTLE)
         self.assertEqual(board.player.movement_mode, MOVES.CASTLE)