Change test input into binrary IO stream

This commit is contained in:
Andy Roberts 2017-01-18 20:51:26 +00:00
parent 486358de69
commit 03a3835e70

View file

@ -86,7 +86,7 @@ class TestEXIFHeader(unittest.TestCase):
def test_with_bytes(self): def test_with_bytes(self):
outputBytes = io.BytesIO() outputBytes = io.BytesIO()
message = b"Secret" message = b"Secret"
exifHeader.hide("./tests/sample-files/20160505T130442.jpg", exifHeader.hide(open("./tests/sample-files/20160505T130442.jpg", 'rb'),
outputBytes, outputBytes,
secret_message=message) secret_message=message)