Piencrypt Test and example

Author: sid86-dev

Test/main.py

@@ -0,0 +1,39 @@
+class PiEncrypt:
+
+    def __init__(self, loc):
+        self.loc = loc
+      
+    # save the data of the picture as bytes 
+    def get_data(self):
+        with open(self.loc , 'rb') as f, open('binary.txt', 'wb') as b:
+            r = f.read()
+            b.write(r)
+
+    # hides the disired data into the picture 
+    def hide_data(self, data):
+        self.data = data
+        with open(self.loc, 'ab') as f:
+            f.write(bytes("HERE" + self.data , encoding="ascii"))
+
+    # read the hidden data from the picture
+    def read_data(self):
+        with open(self.loc, 'rb') as f:
+            content = f.read()
+            list = content.split(b'HERE')
+            # print(f.read())
+            return list[1].decode("utf-8")
+
+    # revert the picture from the backup bytes file
+    def revert(self):
+        with open('binary.txt', 'rb') as f, open(self.loc, 'wb') as e:
+            r = f.read()
+            e.write(r)
+
+if __name__ == '__main__':
+
+    p = PiEncrypt('pic.png')
+    p.get_data()
+    p.hide_data("Hello my name is Sid")
+    r = p.read_data()
+    p.revert()
+    print(r)

Test/pie_clone.py

@@ -0,0 +1,70 @@
+from pie_clone import PiEncrypt
+import random
+import unittest
+
+
+Texts = ['Hi', 'Solo', 'Sid', 'Test', 'Lily', 'Python', 'Throttlerz', 'Password', 'Good Morning', 'Good Evening']   
+
+class TestClac(unittest.TestCase): 
+        def test_quantity(self):  
+            for i in range(800):
+                data = []
+                for i in range(5):
+                    r = random.choice(Texts)
+                    data.append(r)   
+                input = ", ".join(data) 
+                try:
+                    r = PiEncrypt('pic.png')
+                    r.get_data()
+                    r.hide_data(input)
+                    output = r.read_data()
+                    r.revert()
+                except Exception as e:
+                    raise ValueError(e)
+            return output
+
+        def test_quality(self):
+
+            for i in range(10):
+                data = []
+                for i in range(5000):
+                    r = random.choice(Texts)
+                    data.append(r)
+
+                input = ", ".join(data)
+
+                try:
+                    r = PiEncrypt('pic.png')
+                    r.get_data()
+                    r.hide_data(input)
+                    output = r.read_data()
+                    r.revert()
+                except Exception as e:
+                    raise ValueError(e)
+            return output
+
+
+def infile_test():
+    for i in range(10):
+        data = []
+        for i in range(5000):
+            r = random.choice(Texts)
+            data.append(r)
+  
+        input = ", ".join(data)
+        try:
+            r = PiEncrypt('pic.png')
+            r.get_data()
+            r.hide_data(input)
+            output = r.read_data()
+            r.revert()
+            print(output, end=r" %FINISHED% ")
+        except Exception as e:
+            raise ValueError(e)
+  
+if   __name__ == "__main__":
+    unittest.main()
+    # infile_test()
+
+
+