def test_hide_reveal_byte_inverse(data_to_hide): """Anything hidden by _hide_byte is revealed by _reveal_byte.""" clean_data = bytes(b'\x01' * 8) stegs = Steganographer() revealed_byte = stegs._reveal_byte( stegs._hide_byte(clean_data, data_to_hide[0])) assert revealed_byte == data_to_hide
def test_hide_byte(): """The _hide_byte function does hide a byte and returns the test_data with that byte hidden.""" stegs = Steganographer() test_data = bytes(b'\x01' * stegs._BYTELEN) data_to_hide = bytes('A', 'utf-8') solution_data = bytearray(stegs._BYTELEN) solution_data[1] = 1 solution_data[7] = 1 assert stegs._hide_byte(test_data, data_to_hide[0]) == solution_data