from unittest import TestCase
from naja.constants import BITS
-from naja.player import PlayerBits
+from naja.player import PlayerBits, Player
class TestPlayerBits(TestCase):
self.assertEqual(bits._bits, 0x03)
bits.toggle_bits([BITS.NORTH, BITS.CYAN, BITS.MSB])
self.assertEqual(bits._bits, 0x92)
+
+
+class TestPlayer(TestCase):
+ def test_new_player(self):
+ player = Player(0x0f, (0, 1))
+ self.assertEqual(player.bits.bits, 0x0f)
+ self.assertEqual(player.position, (0, 1))
+
+ def test_import_player(self):
+ player = Player.import_player({
+ 'bits': 0xaa,
+ 'position': [1, 2],
+ })
+ self.assertEqual(player.bits.bits, 0xaa)
+ self.assertEqual(player.position, (1, 2))
+
+ def test_export_player(self):
+ player = Player(0xaa, (1, 2))
+ self.assertEqual(player.export(), {
+ 'bits': 0xaa,
+ 'position': [1, 2],
+ })