Ackermann is now encapsulated in a generator.

This commit is contained in:
Cédric Bonhomme 2019-06-01 00:58:00 +02:00
parent 1704aec690
commit 5694e5c806
No known key found for this signature in database
GPG key ID: A1CB94DE57B7A70D
2 changed files with 21 additions and 2 deletions

View file

@ -54,6 +54,18 @@ class TestLSBSet(unittest.TestCase):
generators.eratosthenes())
self.assertEqual(message, clear_message)
def test_hide_and_reveal_with_ackermann(self):
messages_to_hide = ["foo"]
for message in messages_to_hide:
secret = lsbset.hide("./tests/sample-files/Lenna.png", message,
generators.ackermann(m=3))
secret.save("./image.png")
clear_message = lsbset.reveal("./image.png",
generators.ackermann(m=3))
self.assertEqual(message, clear_message)
def test_hide_and_reveal_with_shift(self):
messages_to_hide = ["a", "foo", "Hello World!", ":Python:"]