fz0x1/Stegano
1
0
Fork 0
mirror of https://github.com/cedricbonhomme/Stegano.git synced 2025-05-12 00:58:32 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add test with mersenne generator

Browse source
This commit is contained in:
Flavien 2022-11-11 14:19:34 +01:00
parent 54938159d4
commit 82b59f73db
1 changed files with 14 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
tests/test_lsb.py
View file

@ -90,6 +90,20 @@ class TestLSB(unittest.TestCase):
self.assertEqual(message, clear_message)
def test_hide_and_reveal_with_mersenne(self):
messages_to_hide = ["f"]
for message in messages_to_hide:
secret = lsb.hide(
"./tests/sample-files/Lenna.png",
message,
generators.mersenne(),
)
secret.save("./image.png")
clear_message = lsb.reveal("./image.png", generators.mersenne())
self.assertEqual(message, clear_message)
def test_hide_and_reveal_with_shi_tomashi(self):
messages_to_hide = ["foo bar"]
for message in messages_to_hide: