Change the generator used for the test.

This commit is contained in:
Cédric Bonhomme 2017-06-11 23:40:15 +02:00
parent 26f38f98de
commit 5322ae7839
No known key found for this signature in database
GPG key ID: A1CB94DE57B7A70D

View file

@ -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!"