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
