X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Ftests%2Ftest_player.py;h=07583b3bf11a312bdefed1c267f623b069b42c18;hb=ea9ef7135b38cb5256c0700f5a0ab023d6378831;hp=2e17379213db3d9e4a8f5f6d0603c199ddf88eb8;hpb=266021313a1501c86c11c803851c2b4bb50a8c00;p=naja.git

diff --git a/naja/tests/test_player.py b/naja/tests/test_player.py
index 2e17379..07583b3 100644
--- a/naja/tests/test_player.py
+++ b/naja/tests/test_player.py
@@ -85,6 +85,22 @@ class TestPlayerBits(TestCase):
         bits.toggle_bits([BITS.NORTH, BITS.BLUE, BITS.MSB])
         self.assertEqual(bits._bits, 0x92)
 
+    def test_shift_bits_left(self):
+        bits = PlayerBits(0x03)
+        self.assertEqual(bits._bits, 0x03)
+        bits.shift_bits_left(1)
+        self.assertEqual(bits._bits, 0x06)
+        bits.shift_bits_left(6)
+        self.assertEqual(bits._bits, 0x81)
+
+    def test_shift_bits_right(self):
+        bits = PlayerBits(0x06)
+        self.assertEqual(bits._bits, 0x06)
+        bits.shift_bits_right(1)
+        self.assertEqual(bits._bits, 0x03)
+        bits.shift_bits_right(1)
+        self.assertEqual(bits._bits, 0x81)
+
 
 class TestPlayer(TestCase):
     def test_new_player(self):