Chunk / ImageR

Blame view

test/test_steg.py 2.09 KB
Edit Raw Normal View History
080c30c2   Chunk   F5 lib updated. I... 
1
2
 
__author__ = 'chunk'
84648488   Chunk   reverted. 
3
4
5
6
 
import numpy as np
import pylab as P
import pylab as plt
9ff70cf4   Chunk   capacity engeneer... 
7
 
from .. import mjpeg
84648488   Chunk   reverted. 
8
 
from  ..msteg.steganography import LSB, F3, F4, F5
9ff70cf4   Chunk   capacity engeneer... 
9
 
from ..common import *
080c30c2   Chunk   F5 lib updated. I... 
10
9ff70cf4   Chunk   capacity engeneer... 
11
 
package_dir = os.path.dirname(os.path.abspath(__file__))
080c30c2   Chunk   F5 lib updated. I... 
12
13
14
15
16
17
18
19
20
21
22
23
 

timer = Timer()

sample = [[7, 12, 14, -12, 1, 0, -1, 0],
          [6, 5, -10, 0, 6, 0, 0, 0],
          [0, 6, -5, 4, 0, -1, 0, 0],
          [0, -3, 0, 1, -1, 0, 0, 0],
          [-3, 5, 0, 0, 0, 0, 0, 0],
          [2, -1, 0, 0, 0, 0, 0, 0],
          [0, 0, 0, 0, 0, 0, 0, 0],
          [0, 0, 0, 0, 0, 0, 0, 0]]
84648488   Chunk   reverted. 
24
 
sample_key = [46812L, 20559L, 31360L, 16681L, 27536L, 39553L, 5427L, 63029L, 56572L, 36476L, 25695L, 61908L, 63014L,
080c30c2   Chunk   F5 lib updated. I... 
25
26
 
              5908L, 59816L, 56765L]
84648488   Chunk   reverted. 
27
 
txtsample = [116, 104, 105, 115, 32, 105, 115, 32, 116, 111, 32, 98, 101, 32, 101, 109, 98, 101, 100, 101, 100, 46, 10]
080c30c2   Chunk   F5 lib updated. I... 
28
9ff70cf4   Chunk   capacity engeneer... 
29
30
31
 

def test_F3():
    f3test = F3.F3()
84648488   Chunk   reverted. 
32
 
    f3test.embed_raw_data(os.path.join(package_dir, "../res/50c488a2b163ca8a1f52da6022f03.jpg"), os.path.join(package_dir, "../res/toembed"),
9ff70cf4   Chunk   capacity engeneer... 
33
34
35
36
 
                          os.path.join(package_dir, "../res/steged.jpg"))
    f3test.extract_raw_data(os.path.join(package_dir, "../res/9aa4919175e2d379596fa9947f2e9.jpg"),
                            os.path.join(package_dir, "../res/extracted"))
9ff70cf4   Chunk   capacity engeneer... 
37
38
 
def test_F5():
    f5test = F5.F5(sample_key, 1)
84648488   Chunk   reverted. 
39
 
    f5test.embed_raw_data(os.path.join(package_dir, "../res/50c488a2b163ca8a1f52da6022f03.jpg"), os.path.join(package_dir, "../res/toembed2"),
9ff70cf4   Chunk   capacity engeneer... 
40
41
42
43
44
 
                          os.path.join(package_dir, "../res/steged.jpg"))
    f5test.extract_raw_data(os.path.join(package_dir, "../res/steged.jpg"),
                            os.path.join(package_dir, "../res/extracted"))
84648488   Chunk   reverted. 
45
46
080c30c2   Chunk   F5 lib updated. I... 
47
 
if __name__ == '__main__':
d0be60e7   Chunk   jpeg update. 
48
 
    # f3test = F3.F3()
080c30c2   Chunk   F5 lib updated. I... 
49
50
51
52
53
54
55
56
 
    # f3test = F4.F4(sample_key)
    f3test = F5.F5(sample_key, 1)
    # f3test.embed_raw_data("res/test3.jpg", "res/embeded", "res/steged.jpg")
    f3test.embed_raw_data("res/thulib2.jpg", "res/lena64gray.jpg", "res/steged.jpg")

    # f3test2 = F4.F4(sample_key)
    f3test.extract_raw_data("res/steged.jpg", "res/extracted.jpg")
    print f3test.get_key()
d0be60e7   Chunk   jpeg update. 
57
58
59
 

    # f5test = F5.F5(sample_key, 1)
080c30c2   Chunk   F5 lib updated. I... 
60
 
    pass
84648488   Chunk   reverted.