Initial commit

tests/test_bits.py

@@ -0,0 +1,54 @@
+from unittest import TestCase
+from easybits import Bits
+from easybits.util import is_bit_string
+from easybits.errors import NotEnoughBits, IntegersRequireLength
+
+class TestBits(TestCase):
+    def test_is_bit_string(self):
+        self.assertTrue(is_bit_string('10 10'))
+        self.assertFalse(is_bit_string('T'))
+
+    def test_init_with_bitstring(self):
+        Bits('10101010')
+
+    def test_init_with_bitstring_and_length(self):
+        self.assertEqual(len(Bits('10101010', length=16)), 16)
+        with self.assertRaises(NotEnoughBits):
+            Bits('10101010', length=4)
+
+    def test_init_with_bytes(self):
+        b = Bits(b'a')
+        self.assertEqual(len(b), 8)
+
+    def test_init_with_ascii(self):
+        b = Bits('Chris')
+
+    def test_init_with_utf8(self):
+        b = Bits('😍', encoding='utf8')
+
+    def test_init_with_positive_int(self):
+        b = Bits(12, length=16)
+
+    def test_init_with_negative_int(self):
+        b = Bits(-12, length=16)
+        with self.assertRaises(IntegersRequireLength):
+            Bits(-12)
+
+    def test_int(self):
+        b = Bits(-12, length=16)
+        self.assertEqual(-12, b.int)
+
+    def test_bytes(self):
+        _bytes = '😍'.encode('utf8')
+        b = Bits(_bytes)
+        self.assertEqual(b.bytes, _bytes)
+
+    def test_ascii(self):
+        b = Bits('Chris')
+        self.assertEqual('Chris', b.ascii)
+
+    def test_force_encoded_text(self):
+        b = Bits('1', encoding='ascii')
+        self.assertEqual(len(b), 8)
+
+