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

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
Download patch file Download diff file Expand all files Collapse all files

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()