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