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

Added steganalysis.statistics unit test

Browse source
This commit is contained in:
AdrienCos 2019-06-06 22:36:06 -04:00
parent b054fc7a0c
commit a060757f76
No known key found for this signature in database
GPG key ID: 97AD9AEBCA16D08D
2 changed files with 12 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
tests/expected-results/statistics Normal file
View file

@ -0,0 +1 @@
([58, 56, 54, 57, 60, 59, 62, 61, 63, 65, 64, 66, 67, 254, 68, 69, 255, 70, 253, 71, 72, 74, 252, 73, 251, 250, 75, 249, 76, 77], [(224, 4327), (221, 4138), (223, 4057), (222, 3987), (225, 3713), (220, 3554), (209, 3516), (207, 3476), (208, 3424), (219, 3360)])

12
tests/test_steganalysis.py
View file

@ -32,7 +32,7 @@ import unittest
from unittest.mock import patch
from stegano import lsb
from stegano.steganalysis import parity
from stegano.steganalysis import parity, statistics
from PIL import Image, ImageChops
@ -59,6 +59,16 @@ class TestSteganalysis(unittest.TestCase):
diff = ImageChops.difference(target, analysis).getbbox()
self.assertTrue(diff is None)
def test_statistics(self):
""" Test stegano.steganalysis.statistics
"""
image = Image.open("./tests/sample-files/Lenna.png")
stats = str(statistics.steganalyse(image)) + '\n'
file = open("./tests/expected-results/statistics")
target = file.read()
file.close()
self.assertEqual(stats, target)
if __name__ == '__main__':
unittest.main()