From 5322ae7839d16fbd84212065cc8961209086839b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?C=C3=A9dric=20Bonhomme?= Date: Sun, 11 Jun 2017 23:40:15 +0200 Subject: [PATCH] Change the generator used for the test. --- tests/test_lsbset.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/test_lsbset.py b/tests/test_lsbset.py index 19f45dc..4cd2f50 100644 --- a/tests/test_lsbset.py +++ b/tests/test_lsbset.py @@ -100,9 +100,10 @@ class TestLSBSet(unittest.TestCase): def test_with_too_long_message(self): with open("./tests/sample-files/lorem_ipsum.txt") as f: message = f.read() + message += message*2 with self.assertRaises(Exception): lsbset.hide("./tests/sample-files/Lenna.png", message, - generators.eratosthenes()) + generators.identity()) def test_hide_and_reveal_with_bad_generator(self): message_to_hide = "Hello World!"