def testNegativePos(self):
s = CBS('0xabc', pos=-1)
self.assertEqual(s.pos, 11)
s = CBS('0xabc', pos=-12)
self.assertEqual(s.pos, 0)
with self.assertRaises(bitstring.CreationError):
s = CBS('0xabc', pos=-13)
def testReading(self):
s = CBS(uie=333)
a = s.read('uie')
self.assertEqual(a, 333)
s = CBS('uie=12, sie=-9, sie=9, uie=1000000')
u = s.unpack('uie, 2*sie, uie')
self.assertEqual(u, [12, -9, 9, 1000000])
def testExceptions(self):
f = io.BytesIO(b"123456789")
s = CBS(f, length=9 * 8)
with self.assertRaises(bitstring.CreationError):
s = CBS(f, length=9 * 8 + 1)
with self.assertRaises(bitstring.CreationError):
s = CBS(f, length=9 * 8, offset=1)
def testSimpleCreation(self):
f = io.BytesIO(b"\x12\xff\x77helloworld")
s = CBS(f)
self.assertEqual(s[0:8], '0x12')
self.assertEqual(s.len, 13 * 8)
s = CBS(f, offset=8, length=12)
self.assertEqual(s, '0xff7')
def testPositivePos(self):
s = CBS('0xabc', pos=0)
self.assertEqual(s.pos, 0)
s = CBS('0xabc', pos=1)
self.assertEqual(s.pos, 1)
s = CBS('0xabc', pos=12)
self.assertEqual(s.pos, 12)
with self.assertRaises(bitstring.CreationError):
s = CBS('0xabc', pos=13)
def testRead(self):
s = CBS('0b100011110001')
a = s.read('pad:1')
self.assertEqual(a, None)
self.assertEqual(s.pos, 1)
a = s.read(3)
self.assertEqual(a, CBS('0b000'))
a = s.read('pad:0')
self.assertEqual(a, None)
self.assertEqual(s.pos, 4)
def testNotAligned(self):
a = CBS('0b00111001001010011011')
a.pos = 1
self.assertEqual(a.readto('0b00'), '0b011100')
self.assertEqual(a.readto('0b110'), '0b10010100110')
with self.assertRaises(ValueError):
a.readto('')
def testFromFile(self):
s = CBS(filename='test.m1v')
self.assertEqual(s[0:32].hex, '000001b3')
self.assertEqual(s.read(8 * 4).hex, '000001b3')
width = s.read(12).uint
height = s.read(12).uint
self.assertEqual((width, height), (352, 288))
def testReadingLines(self):
s = b"This is a test\nof reading lines\nof text\n"
b = CBS(bytes=s)
n = bitstring.Bits(bytes=b'\n')
self.assertEqual(b.readto(n).bytes, b'This is a test\n')
self.assertEqual(b.readto(n).bytes, b'of reading lines\n')
self.assertEqual(b.readto(n).bytes, b'of text\n')
def testByteAligned(self):
a = CBS('0xaabb00aa00bb')
b = a.readto('0x00', bytealigned=True)
self.assertEqual(b, '0xaabb00')
self.assertEqual(a.bytepos, 3)
b = a.readto('0xaa', bytealigned=True)
self.assertEqual(b, '0xaa')
self.assertRaises(bitstring.ReadError, a.readto, '0xcc', bytealigned=True)
def testUnpackingBytes(self):
s = CBS(80)
t = s.unpack('bytes:1')
self.assertEqual(t[0], b'\x00')
a, b, c = s.unpack('bytes:1, bytes, bytes:2')
self.assertEqual(a, b'\x00')
self.assertEqual(b, b'\x00' * 7)
self.assertEqual(c, b'\x00' * 2)
def testReadingErrors(self):
s = CBS(10)
with self.assertRaises(bitstring.ReadError):
s.read('uie')
self.assertEqual(s.pos, 0)
with self.assertRaises(bitstring.ReadError):
s.read('sie')
self.assertEqual(s.pos, 0)
def testStringRepresentationFromFile(self):
s = CBS(filename='test.m1v', pos=2001)
self.assertEqual(
s.__repr__(),
"ConstBitStream(filename='test.m1v', length=1002400, pos=2001)")
s.pos = 0
self.assertEqual(
s.__repr__(),
"ConstBitStream(filename='test.m1v', length=1002400)")
def testReadList(self):
s = CBS('0b10001111001')
t = s.readlist('pad:1, uint:3, pad:4, uint:3')
self.assertEqual(t, [0, 1])
s.pos = 0
t = s.readlist('pad:1, pad:5')
self.assertEqual(t, [])
self.assertEqual(s.pos, 6)
s.pos = 0
t = s.readlist('pad:1, bin, pad:4, uint:3')
self.assertEqual(t, ['000', 1])
s.pos = 0
t = s.readlist('pad, bin:3, pad:4, uint:3')
self.assertEqual(t, ['000', 1])
def testDisallowIntegers(self):
a = CBS('0x0f')
self.assertRaises(ValueError, a.readto, 4)
def testDefaultCreation(self):
s = CBS('0xabc')
self.assertEqual(s.pos, 0)
def testStringRepresentation(self):
s = CBS('0b110', pos=2)
self.assertEqual(s.__repr__(), "ConstBitStream('0b110', pos=2)")
def testUnpackingBytesWithKeywords(self):
s = CBS('0x55' * 10)
t = s.unpack('pad:a, bytes:b, bytes, pad:a', a=4, b=6)
self.assertEqual(t, [b'\x55' * 6, b'\x55' * 3])
def testDisallowIntegers(self):
a = CBS('0x0f')
with self.assertRaises(ValueError):
a.readto(4)
def testReadingHex(self):
s = CBS('0xabcdef')
self.assertEqual(s.read(4), '0xf')
self.assertEqual(s.read(4), '0xe')
self.assertEqual(s.pos, 8)