X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Ftests%2Ftest_player.py;h=d2b896e402b3983c6b594d5e3c12c24f0304b482;hb=18eaa15b16160df3dcd23dd2773603d1f2e179ce;hp=8373eee2d53a5cc35d9049854646bf6a0ea947d8;hpb=eccb327f1260b9aa981755840e097d5b23af05a9;p=naja.git diff --git a/naja/tests/test_player.py b/naja/tests/test_player.py index 8373eee..d2b896e 100644 --- a/naja/tests/test_player.py +++ b/naja/tests/test_player.py @@ -52,35 +52,35 @@ class TestPlayerBits(TestCase): def test_check_bits(self): bits = PlayerBits(0x03) - self.assertEqual(bits.check_bits(BITS.NORTH), True) - self.assertEqual(bits.check_bits(BITS.NORTH, BITS.SOUTH), True) - self.assertEqual(bits.check_bits(BITS.CYAN), False) - self.assertEqual(bits.check_bits(BITS.CYAN, BITS.MSB), False) + self.assertEqual(bits.check_bits([BITS.NORTH]), True) + self.assertEqual(bits.check_bits([BITS.NORTH, BITS.SOUTH]), True) + self.assertEqual(bits.check_bits([BITS.CYAN]), False) + self.assertEqual(bits.check_bits([BITS.CYAN, BITS.MSB]), False) self.assertEqual( - bits.check_bits(BITS.NORTH, BITS.SOUTH, BITS.CYAN), False) + bits.check_bits([BITS.NORTH, BITS.SOUTH, BITS.CYAN]), False) def test_set_bits(self): bits = PlayerBits(0x03) self.assertEqual(bits._bits, 0x03) - bits.set_bits(BITS.NORTH, BITS.CYAN) + bits.set_bits([BITS.NORTH, BITS.CYAN]) self.assertEqual(bits._bits, 0x13) - bits.set_bits(BITS.NORTH, BITS.CYAN, BITS.MSB) + bits.set_bits([BITS.NORTH, BITS.CYAN, BITS.MSB]) self.assertEqual(bits._bits, 0x93) def test_clear_bits(self): bits = PlayerBits(0x03) self.assertEqual(bits._bits, 0x03) - bits.clear_bits(BITS.NORTH, BITS.CYAN) + bits.clear_bits([BITS.NORTH, BITS.CYAN]) self.assertEqual(bits._bits, 0x02) - bits.clear_bits(BITS.NORTH, BITS.CYAN, BITS.MSB) + bits.clear_bits([BITS.NORTH, BITS.CYAN, BITS.MSB]) self.assertEqual(bits._bits, 0x02) def test_toggle_bits(self): bits = PlayerBits(0x03) self.assertEqual(bits._bits, 0x03) - bits.toggle_bits(BITS.NORTH, BITS.CYAN) + bits.toggle_bits([BITS.NORTH, BITS.CYAN]) self.assertEqual(bits._bits, 0x12) - bits.toggle_bits(BITS.NORTH, BITS.CYAN) + bits.toggle_bits([BITS.NORTH, BITS.CYAN]) self.assertEqual(bits._bits, 0x03) - bits.toggle_bits(BITS.NORTH, BITS.CYAN, BITS.MSB) + bits.toggle_bits([BITS.NORTH, BITS.CYAN, BITS.MSB]) self.assertEqual(bits._bits, 0x92)