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_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"))
