aboutsummaryrefslogtreecommitdiff
path: root/tests/test_gameBoard.py
blob: 2fa10370bc69ad7e2f8d0a110f108025472d8db4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
"""Tests for gameBoard module."""

import unittest

from imago.data.enums import Player
from imago.gameLogic.gameBoard import GameBoard

#from imago.data.enums import Player

TEST_BOARD_SIZE = 19

class TestGameBoard(unittest.TestCase):
    """Test gameBoard module."""

    def testGetGroupLiberties(self):
        """Test calculation of group liberties."""
        board = GameBoard(TEST_BOARD_SIZE)

        #Empty cell liberties
        self.assertEqual(board.getGroupLiberties(0,0), {})
        self.assertEqual(board.getGroupLibertiesCount(0,0), 0)

        # Lone stone liberties
        board.board[3][3] = Player.WHITE
        self.assertEqual(board.getGroupLiberties(3,3),
                {(2,3), (3,2), (4,3), (3,4)})
        self.assertEqual(board.getGroupLibertiesCount(3,3), 4)

if __name__ == '__main__':
    unittest.main()