1 from unittest import TestCase
3 from naja.constants import BITS
4 from naja.player import PlayerBits
7 class TestPlayerBits(TestCase):
8 def test_range_limits(self):
9 bits = PlayerBits(0x00)
14 self.assertEqual(bits.bits, 0x00)
15 self.assertRaises(AssertionError, set_value, -1)
16 self.assertEqual(bits.bits, 0x00)
18 self.assertEqual(bits.bits, 0xff)
20 self.assertEqual(bits.bits, 0x00)
21 self.assertRaises(AssertionError, set_value, 256)
22 self.assertEqual(bits.bits, 0x00)
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)
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)
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)
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)
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)
60 bits.check_bits(BITS.NORTH, BITS.SOUTH, BITS.CYAN), False)
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)
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)
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)