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         })