Factor out EIGHT_BIT_SCALE.

[naja.git] / naja / tests / test_gameboard.py

diff --git a/naja/tests/test_gameboard.py b/naja/tests/test_gameboard.py
index 0ad0d28a5f218df26b9be12554094bd6a880d1ec..9d39fe85a4e34a2fcb7a5ad3290ce4104ad5ed7f 100644 (file)
--- a/naja/tests/test_gameboard.py
+++ b/naja/tests/test_gameboard.py
@@ -2,14 +2,44 @@ from unittest import TestCase
 
 from naja.constants import BITS
 from naja.gameboard import GameBoard, LocationCard
+from naja import actions
 
 
 class TestGameBoard(TestCase):
+    def test_export_new_board(self):
+        board = GameBoard.new_game([{'actions': []}])
+        exported_state = board.export()
+        board_locations = exported_state.pop('board_locations')
+        self.assertEqual(exported_state, {
+            'max_health': 4,
+            'health': 4,
+            'wins_required': 4,
+            'wins': 0,
+            'locations': [{'actions': []}],
+            '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():
+            self.assertEqual(
+                sorted(location_state.keys()), ['actions', 'bitwise_operand'])
+            self.assertEqual(location_state['actions'], [])
+            self.assertTrue(2 <= len(location_state['bitwise_operand']) <= 3)
+
     def test_lose_health(self):
-        board = GameBoard(None, 4, 4, locations={}, state=None)
+        board = GameBoard.new_game([{'actions': []}])
         self.assertEqual(board.health, 4)
+        state_1 = board.export()
+
         board.lose_health()
         self.assertEqual(board.health, 3)
+        state_2 = board.export()
+
+        # Make sure nothing else has changed.
+        state_1.pop('health')
+        state_2.pop('health')
+        self.assertEqual(state_1, state_2)
 
 
 class TestLocationCard(TestCase):
@@ -28,3 +58,15 @@ class TestLocationCard(TestCase):
         self.assertTrue(sizes[3] > 0)
         self.assertTrue(sizes[2] > sizes[3])
         self.assertEqual(bits, set(BITS.values()))
+
+    def test_new_location_no_actions(self):
+        location = LocationCard.new_location({'actions': []})
+        self.assertEqual(location.actions, [])
+
+    def test_new_location_one_action(self):
+        location = LocationCard.new_location({'actions': [
+            {'required_bits': [], 'action_class': 'DoNothing'},
+        ]})
+        [action] = location.actions
+        self.assertEqual(type(action), actions.DoNothing)
+        self.assertEqual(action.required_bits, set())