naja/tests/test_player.py

   1 from unittest import TestCase
   2
   3 from naja.constants import BITS, MOVES
   4 from naja.player import PlayerBits, Player
   5
   6
   7 class TestPlayerBits(TestCase):
   8     def test_range_limits(self):
   9         bits = PlayerBits(0x00)
  10
  11         def set_value(value):
  12             bits.bits = value
  13
  14         self.assertEqual(bits.bits, 0x00)
  15         self.assertRaises(AssertionError, set_value, -1)
  16         self.assertEqual(bits.bits, 0x00)
  17         bits.bits = 255
  18         self.assertEqual(bits.bits, 0xff)
  19         bits.bits = 0
  20         self.assertEqual(bits.bits, 0x00)
  21         self.assertRaises(AssertionError, set_value, 256)
  22         self.assertEqual(bits.bits, 0x00)
  23
  24     def test_check_bit(self):
  25         bits = PlayerBits(0x01)
  26         self.assertEqual(bits.check_bit(BITS.NORTH), True)
  27         self.assertEqual(bits.check_bit(BITS.SOUTH), False)
  28
  29     def test_set_bit(self):
  30         bits = PlayerBits(0x00)
  31         self.assertEqual(bits._bits, 0x00)
  32         bits.set_bit(BITS.NORTH)
  33         self.assertEqual(bits._bits, 0x01)
  34         bits.set_bit(BITS.NORTH)
  35         self.assertEqual(bits._bits, 0x01)
  36
  37     def test_clear_bit(self):
  38         bits = PlayerBits(0x01)
  39         self.assertEqual(bits._bits, 0x01)
  40         bits.clear_bit(BITS.NORTH)
  41         self.assertEqual(bits._bits, 0x00)
  42         bits.clear_bit(BITS.NORTH)
  43         self.assertEqual(bits._bits, 0x00)
  44
  45     def test_toggle_bit(self):
  46         bits = PlayerBits(0x00)
  47         self.assertEqual(bits._bits, 0x00)
  48         bits.toggle_bit(BITS.NORTH)
  49         self.assertEqual(bits._bits, 0x01)
  50         bits.toggle_bit(BITS.NORTH)
  51         self.assertEqual(bits._bits, 0x00)
  52
  53     def test_check_bits(self):
  54         bits = PlayerBits(0x03)
  55         self.assertEqual(bits.check_bits([BITS.NORTH]), True)
  56         self.assertEqual(bits.check_bits([BITS.NORTH, BITS.SOUTH]), True)
  57         self.assertEqual(bits.check_bits([BITS.BLUE]), False)
  58         self.assertEqual(bits.check_bits([BITS.BLUE, BITS.MSB]), False)
  59         self.assertEqual(
  60             bits.check_bits([BITS.NORTH, BITS.SOUTH, BITS.BLUE]), False)
  61
  62     def test_set_bits(self):
  63         bits = PlayerBits(0x03)
  64         self.assertEqual(bits._bits, 0x03)
  65         bits.set_bits([BITS.NORTH, BITS.BLUE])
  66         self.assertEqual(bits._bits, 0x13)
  67         bits.set_bits([BITS.NORTH, BITS.BLUE, BITS.MSB])
  68         self.assertEqual(bits._bits, 0x93)
  69
  70     def test_clear_bits(self):
  71         bits = PlayerBits(0x03)
  72         self.assertEqual(bits._bits, 0x03)
  73         bits.clear_bits([BITS.NORTH, BITS.BLUE])
  74         self.assertEqual(bits._bits, 0x02)
  75         bits.clear_bits([BITS.NORTH, BITS.BLUE, BITS.MSB])
  76         self.assertEqual(bits._bits, 0x02)
  77
  78     def test_toggle_bits(self):
  79         bits = PlayerBits(0x03)
  80         self.assertEqual(bits._bits, 0x03)
  81         bits.toggle_bits([BITS.NORTH, BITS.BLUE])
  82         self.assertEqual(bits._bits, 0x12)
  83         bits.toggle_bits([BITS.NORTH, BITS.BLUE])
  84         self.assertEqual(bits._bits, 0x03)
  85         bits.toggle_bits([BITS.NORTH, BITS.BLUE, BITS.MSB])
  86         self.assertEqual(bits._bits, 0x92)
  87
  88
  89 class TestPlayer(TestCase):
  90     def test_new_player(self):
  91         player = Player(0x0f, (0, 1))
  92         self.assertEqual(player.bits.bits, 0x0f)
  93         self.assertEqual(player.position, (0, 1))
  94
  95     def test_import_player(self):
  96         player = Player.import_player({
  97             'bits': 0xaa,
  98             'position': [1, 2],
  99             'movement_mode': MOVES.CASTLE,
 100         })
 101         self.assertEqual(player.bits.bits, 0xaa)
 102         self.assertEqual(player.position, (1, 2))
 103         self.assertEqual(player.movement_mode, MOVES.CASTLE)
 104
 105     def test_export_player(self):
 106         player = Player(0xaa, (1, 2), MOVES.CASTLE)
 107         self.assertEqual(player.export(), {
 108             'bits': 0xaa,
 109             'position': [1, 2],
 110             'movement_mode': MOVES.CASTLE,
 111         })
 112
 113     def test_export_player_default_movement_mode(self):
 114         player = Player(0xaa, (1, 2))
 115         self.assertEqual(player.export(), {
 116             'bits': 0xaa,
 117             'position': [1, 2],
 118             'movement_mode': MOVES.ADJACENT,
 119         })
 120
 121     def test_legal_moves_all_available(self):
 122         player = Player(0x0f, (2, 2))
 123         self.assertEqual(
 124             player.legal_moves(), [(2, 2), (2, 1), (2, 3), (3, 2), (1, 2)])
 125
 126     def test_legal_moves_some_unavailable(self):
 127         player = Player(0x0f, (0, 2))
 128         player.bits.clear_bit(BITS.NORTH)
 129         self.assertEqual(player.legal_moves(), [(0, 2), (0, 3), (1, 2)])
 130
 131     def test_legal_moves_castle(self):
 132         player = Player(0x0f, (1, 3), MOVES.CASTLE)
 133         self.assertEqual(player.legal_moves(), [
 134             (0, 3), (1, 0), (1, 1), (1, 2), (1, 3), (1, 4), (2, 3), (3, 3),
 135             (4, 3)])
 136
 137     def test_legal_moves_bishop(self):
 138         player = Player(0x0f, (1, 3), MOVES.BISHOP)
 139         self.assertEqual(player.legal_moves(), [
 140             (0, 2), (0, 4), (1, 3), (2, 2), (2, 4), (3, 1), (4, 0)])
 141
 142     def test_legal_moves_knight(self):
 143         player = Player(0x0f, (1, 3), MOVES.KNIGHT)
 144         self.assertEqual(player.legal_moves(), [
 145             (0, 1), (1, 3), (2, 1), (3, 2), (3, 4)])
 146
 147     def test_set_position(self):
 148         player = Player(0x0f, (3, 3), MOVES.BISHOP)
 149         player.set_position((4, 4))
 150         self.assertEqual(player.movement_mode, MOVES.ADJACENT)