From 655d211dc02e0057b34e8d76df6fbb03a7a34c77 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?C=C3=A9dric=20Bonhomme?= Date: Wed, 13 Apr 2016 08:11:12 +0200 Subject: [PATCH] Bug fix: forgot to write the results in a file. --- tests/test_slsb.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/test_slsb.py b/tests/test_slsb.py index 0c7c63a..5fb935b 100644 --- a/tests/test_slsb.py +++ b/tests/test_slsb.py @@ -57,6 +57,10 @@ class TestSLSB(unittest.TestCase): with open("./examples/lorem_ipsum.txt") as f: message = f.read() secret = slsb.hide("./examples/pictures/Lenna.png", message) + secret.save("./image.png") + + clear_message = slsb.reveal("./image.png") + self.assertEqual(message, clear_message) def test_with_too_long_message(self): with open("./examples/lorem_ipsum.txt") as f: