naja/tests/test_player.py

   1 from unittest import TestCase
   2
   3 from naja.constants import BITS
   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.CYAN]), False)
  58         self.assertEqual(bits.check_bits([BITS.CYAN, BITS.MSB]), False)
  59         self.assertEqual(
  60             bits.check_bits([BITS.NORTH, BITS.SOUTH, BITS.CYAN]), 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.CYAN])
  66         self.assertEqual(bits._bits, 0x13)
  67         bits.set_bits([BITS.NORTH, BITS.CYAN, 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.CYAN])
  74         self.assertEqual(bits._bits, 0x02)
  75         bits.clear_bits([BITS.NORTH, BITS.CYAN, 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.CYAN])
  82         self.assertEqual(bits._bits, 0x12)
  83         bits.toggle_bits([BITS.NORTH, BITS.CYAN])
  84         self.assertEqual(bits._bits, 0x03)
  85         bits.toggle_bits([BITS.NORTH, BITS.CYAN, 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         })
 100         self.assertEqual(player.bits.bits, 0xaa)
 101         self.assertEqual(player.position, (1, 2))
 102
 103     def test_export_player(self):
 104         player = Player(0xaa, (1, 2))
 105         self.assertEqual(player.export(), {
 106             'bits': 0xaa,
 107             'position': [1, 2],
 108         })
 109
 110     def test_move(self):
 111         player = Player(0x0f, (2, 2))
 112         self.assertEqual(player.move(BITS.NORTH), True)
 113         self.assertEqual(player.position, (2, 1))
 114
 115         player = Player(0x0f, (2, 2))
 116         self.assertEqual(player.move(BITS.SOUTH), True)
 117         self.assertEqual(player.position, (2, 3))
 118
 119         player = Player(0x0f, (2, 2))
 120         self.assertEqual(player.move(BITS.WEST), True)
 121         self.assertEqual(player.position, (1, 2))
 122
 123         player = Player(0x0f, (2, 2))
 124         self.assertEqual(player.move(BITS.EAST), True)
 125         self.assertEqual(player.position, (3, 2))
 126
 127     def test_move_flags_clear(self):
 128         player = Player(0x00, (2, 2))
 129         self.assertEqual(player.move(BITS.NORTH), False)
 130         self.assertEqual(player.position, (2, 2))
 131
 132         player = Player(0x00, (2, 2))
 133         self.assertEqual(player.move(BITS.SOUTH), False)
 134         self.assertEqual(player.position, (2, 2))
 135
 136         player = Player(0x00, (2, 2))
 137         self.assertEqual(player.move(BITS.WEST), False)
 138         self.assertEqual(player.position, (2, 2))
 139
 140         player = Player(0x00, (2, 2))
 141         self.assertEqual(player.move(BITS.EAST), False)
 142         self.assertEqual(player.position, (2, 2))
 143
 144     def test_move_flags_edges(self):
 145         player = Player(0x0f, (2, 0))
 146         self.assertEqual(player.move(BITS.NORTH), False)
 147         self.assertEqual(player.position, (2, 0))
 148
 149         player = Player(0x0f, (2, 4))
 150         self.assertEqual(player.move(BITS.SOUTH), False)
 151         self.assertEqual(player.position, (2, 4))
 152
 153         player = Player(0x0f, (0, 2))
 154         self.assertEqual(player.move(BITS.WEST), False)
 155         self.assertEqual(player.position, (0, 2))
 156
 157         player = Player(0x0f, (4, 2))
 158         self.assertEqual(player.move(BITS.EAST), False)
 159         self.assertEqual(player.position, (4, 2))
 160
 161     def test_legal_moves_all_available(self):
 162         player = Player(0x0f, (2, 2))
 163         self.assertEqual(
 164             player.legal_moves(), [(2, 2), (2, 1), (2, 3), (3, 2), (1, 2)])
 165
 166     def test_legal_moves_some_unavailable(self):
 167         player = Player(0x0f, (0, 2))
 168         player.bits.clear_bit(BITS.NORTH)
 169         self.assertEqual(player.legal_moves(), [(0, 2), (0, 3), (1, 2)])