fz0x1/Stegano
1
0
Fork 0
mirror of https://github.com/cedricbonhomme/Stegano.git synced 2025-06-28 11:16:14 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

added a shift parameter for the lsbset module.

Browse source
This commit is contained in:
Cédric Bonhomme 2018-12-18 21:38:25 +01:00
parent a86361dd27
commit ea2de8142c
No known key found for this signature in database
GPG key ID: A1CB94DE57B7A70D
6 changed files with 47 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

16
tests/test_lsbset.py
View file

@ -53,17 +53,29 @@ class TestLSBSet(unittest.TestCase):
generators.eratosthenes())
self.assertEqual(message, clear_message)
def test_hide_and_reveal_with_shift(self):
messages_to_hide = ["a", "foo", "Hello World!", ":Python:"]
for message in messages_to_hide:
secret = lsbset.hide("./tests/sample-files/Lenna.png", message,
generators.eratosthenes(), 4)
secret.save("./image.png")
clear_message = lsbset.reveal("./image.png",
generators.eratosthenes(), 4)
self.assertEqual(message, clear_message)
def test_hide_and_reveal_UTF32LE(self):
messages_to_hide = 'I love 🍕 and 🍫!'
secret = lsbset.hide("./tests/sample-files/Lenna.png",
messages_to_hide,
generators.eratosthenes(),
'UTF-32LE')
encoding='UTF-32LE')
secret.save("./image.png")
clear_message = lsbset.reveal("./image.png", generators.eratosthenes(),
'UTF-32LE')
encoding='UTF-32LE')
self.assertEqual(messages_to_hide, clear_message)
def test_with_transparent_png(self):