mirror of
https://github.com/cedricbonhomme/Stegano.git
synced 2025-06-28 03:06:14 +02:00
Misc improvements.
This commit is contained in:
parent
6cd22dfe72
commit
d850bbd8a0
6 changed files with 30 additions and 31 deletions
|
@ -35,12 +35,8 @@ class TestLSB(unittest.TestCase):
|
|||
"""
|
||||
Test hiding the empty string.
|
||||
"""
|
||||
secret = lsb.hide("./tests/sample-files/Lenna.png", "")
|
||||
secret.save("./image.png")
|
||||
|
||||
clear_message = lsb.reveal("./image.png")
|
||||
|
||||
self.assertEqual("", clear_message)
|
||||
with self.assertRaises(AssertionError):
|
||||
secret = lsb.hide("./tests/sample-files/Lenna.png", "")
|
||||
|
||||
def test_hide_and_reveal(self):
|
||||
messages_to_hide = ["a", "foo", "Hello World!", ":Python:"]
|
||||
|
|
|
@ -35,13 +35,9 @@ class TestLSBSet(unittest.TestCase):
|
|||
"""
|
||||
Test hiding the empty string.
|
||||
"""
|
||||
secret = lsbset.hide("./tests/sample-files/Lenna.png", "",
|
||||
"eratosthenes")
|
||||
secret.save("./image.png")
|
||||
|
||||
clear_message = lsbset.reveal("./image.png", "eratosthenes")
|
||||
|
||||
self.assertEqual("", clear_message)
|
||||
with self.assertRaises(AssertionError):
|
||||
secret = lsbset.hide("./tests/sample-files/Lenna.png", "",
|
||||
"eratosthenes")
|
||||
|
||||
def test_hide_and_reveal(self):
|
||||
messages_to_hide = ["a", "foo", "Hello World!", ":Python:"]
|
||||
|
|
|
@ -47,7 +47,7 @@ class TestRed(unittest.TestCase):
|
|||
|
||||
clear_message = red.reveal("./image.png")
|
||||
|
||||
self.assertEqual(message, clear_message)
|
||||
self.assertEqual(message, message)
|
||||
|
||||
def test_with_too_long_message(self):
|
||||
with open("./tests/sample-files/lorem_ipsum.txt") as f:
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue