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

Reorganization of all modules.

Browse source
This commit is contained in:
Cédric Bonhomme 2016-05-19 21:32:15 +02:00
parent e0bed8ba52
commit 872a5546fc
No known key found for this signature in database
GPG key ID: A1CB94DE57B7A70D
23 changed files with 77 additions and 43 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/test_slsb.py → tests/test_lsb.py
View file

@ -27,18 +27,18 @@ __license__ = "GPLv3"
import os
import unittest
from stegano import slsb
from stegano import lsb
class TestSLSB(unittest.TestCase):
class TestLSB(unittest.TestCase):
def test_hide_empty_message(self):
"""
Test hiding the empty string.
"""
secret = slsb.hide("./examples/pictures/Lenna.png", "")
secret = lsb.hide("./examples/pictures/Lenna.png", "")
secret.save("./image.png")
clear_message = slsb.reveal("./image.png")
clear_message = lsb.reveal("./image.png")
self.assertEqual("", clear_message)
@ -46,20 +46,20 @@ class TestSLSB(unittest.TestCase):
messages_to_hide = ["a", "foo", "Hello World!", ":Python:"]
for message in messages_to_hide:
secret = slsb.hide("./examples/pictures/Lenna.png", message)
secret = lsb.hide("./examples/pictures/Lenna.png", message)
secret.save("./image.png")
clear_message = slsb.reveal("./image.png")
clear_message = lsb.reveal("./image.png")
self.assertEqual(message, clear_message)
def test_with_long_message(self):
with open("./examples/lorem_ipsum.txt") as f:
message = f.read()
secret = slsb.hide("./examples/pictures/Lenna.png", message)
secret = lsb.hide("./examples/pictures/Lenna.png", message)
secret.save("./image.png")
clear_message = slsb.reveal("./image.png")
clear_message = lsb.reveal("./image.png")
self.assertEqual(message, clear_message)
def test_with_too_long_message(self):
@ -67,7 +67,7 @@ class TestSLSB(unittest.TestCase):
message = f.read()
message += message*2
with self.assertRaises(Exception):
slsb.hide("./examples/pictures/Lenna.png", message)
lsb.hide("./examples/pictures/Lenna.png", message)
def tearDown(self):
try:

16
tests/test_slsbset.py → tests/test_lsbset.py
View file

@ -27,19 +27,19 @@ __license__ = "GPLv3"
import os
import unittest
from stegano import slsbset
from stegano import lsbset
class TestSLSBSet(unittest.TestCase):
class TestLSBSet(unittest.TestCase):
def test_hide_empty_message(self):
"""
Test hiding the empty string.
"""
secret = slsbset.hide("./examples/pictures/Lenna.png", "",
secret = lsbset.hide("./examples/pictures/Lenna.png", "",
"eratosthenes")
secret.save("./image.png")
clear_message = slsbset.reveal("./image.png", "eratosthenes")
clear_message = lsbset.reveal("./image.png", "eratosthenes")
self.assertEqual("", clear_message)
@ -47,23 +47,23 @@ class TestSLSBSet(unittest.TestCase):
messages_to_hide = ["a", "foo", "Hello World!", ":Python:"]
for message in messages_to_hide:
secret = slsbset.hide("./examples/pictures/Lenna.png", message,
secret = lsbset.hide("./examples/pictures/Lenna.png", message,
"eratosthenes")
secret.save("./image.png")
clear_message = slsbset.reveal("./image.png", "eratosthenes")
clear_message = lsbset.reveal("./image.png", "eratosthenes")
self.assertEqual(message, clear_message)
def test_hide_and_reveal_with_bad_generator(self):
message_to_hide = "Hello World!"
secret = slsbset.hide("./examples/pictures/Lenna.png", message_to_hide,
secret = lsbset.hide("./examples/pictures/Lenna.png", message_to_hide,
"eratosthenes")
secret.save("./image.png")
with self.assertRaises(IndexError):
clear_message = slsbset.reveal("./image.png", "identity")
clear_message = lsbset.reveal("./image.png", "identity")
def tearDown(self):
try: