def test_bits_read2_padded2(self):
stream = six.BytesIO(pfp.utils.binary(chr(int("11110000",2)) + chr(int("10101010", 2))))
bitwrapped = BitwrappedStream(stream)
bitwrapped.padded = True
res = bitwrapped.read_bits(4)
self.assertEqual([1,1,1,1], res)
next_byte = bitwrapped.read(1)
self.assertEqual(pfp.utils.binary(chr(int("10101010", 2))), next_byte)
def test_bits_write_padded(self):
stream = six.BytesIO()
bitwrapped = BitwrappedStream(stream)
bitwrapped.padded = True
bitwrapped.write_bits([1,1,0,1])
# should go to a new byte now, zero padded after the
# 1101 bits
bitwrapped.write(pfp.utils.binary("hello"))
self.assertEqual(stream.getvalue(), pfp.utils.binary(chr(int("11010000", 2)) + "hello"))
def test_bits_read2_padded2(self):
stream = six.BytesIO(
pfp.utils.binary(
chr(int("11110000", 2)) + chr(int("10101010", 2))))
bitwrapped = BitwrappedStream(stream)
bitwrapped.padded = True
res = bitwrapped.read_bits(4)
self.assertEqual([1, 1, 1, 1], res)
next_byte = bitwrapped.read(1)
self.assertEqual(pfp.utils.binary(chr(int("10101010", 2))), next_byte)
def test_bits_write_padded(self):
stream = six.BytesIO()
bitwrapped = BitwrappedStream(stream)
bitwrapped.padded = True
bitwrapped.write_bits([1, 1, 0, 1])
# should go to a new byte now, zero padded after the
# 1101 bits
bitwrapped.write(pfp.utils.binary("hello"))
self.assertEqual(stream.getvalue(),
pfp.utils.binary(chr(int("11010000", 2)) + "hello"))
def test_bits_read_unpadded(self):
stream = six.BytesIO(pfp.utils.binary(chr(int("11110000",2)) + chr(int("10101010", 2))))
bitwrapped = BitwrappedStream(stream)
bitwrapped.padded = False
res = bitwrapped.read_bits(4)
self.assertEqual([1,1,1,1], res)
res = bitwrapped.read(1)
self.assertEqual(pfp.utils.binary(chr(int("00001010", 2))), res)
res = bitwrapped.read_bits(4)
self.assertEqual([1,0,1,0], res)
def test_bits_read_unpadded(self):
stream = six.BytesIO(
pfp.utils.binary(
chr(int("11110000", 2)) + chr(int("10101010", 2))))
bitwrapped = BitwrappedStream(stream)
bitwrapped.padded = False
res = bitwrapped.read_bits(4)
self.assertEqual([1, 1, 1, 1], res)
res = bitwrapped.read(1)
self.assertEqual(pfp.utils.binary(chr(int("00001010", 2))), res)
res = bitwrapped.read_bits(4)
self.assertEqual([1, 0, 1, 0], res)
def test_bits_read2_padded1(self):
stream = six.BytesIO(pfp.utils.binary(chr(int("11110000",2)) + chr(int("10101010", 2))))
bitwrapped = BitwrappedStream(stream)
bitwrapped.padded = True
res = bitwrapped.read_bits(4)
self.assertEqual([1,1,1,1], res)
res = bitwrapped.read_bits(3)
self.assertEqual([0,0,0], res)
res = bitwrapped.read_bits(4)
self.assertEqual([0,1,0,1], res)
res = bitwrapped.read_bits(5)
self.assertEqual([0,1,0,1,0], res)
def test_bits_read2_padded1(self):
stream = six.BytesIO(
pfp.utils.binary(
chr(int("11110000", 2)) + chr(int("10101010", 2))))
bitwrapped = BitwrappedStream(stream)
bitwrapped.padded = True
res = bitwrapped.read_bits(4)
self.assertEqual([1, 1, 1, 1], res)
res = bitwrapped.read_bits(3)
self.assertEqual([0, 0, 0], res)
res = bitwrapped.read_bits(4)
self.assertEqual([0, 1, 0, 1], res)
res = bitwrapped.read_bits(5)
self.assertEqual([0, 1, 0, 1, 0], res)
