added save_input unittest

win_venv/main.py

@@ -23,9 +23,8 @@ def save_input(oldpath):
     timestamp = str(int(time.time()))
     filename = timestamp + '.' + oldpath.split('.')[-1]
     newpath = filePath + '\\' + filename
-    print(oldpath)
-    print(newpath)
     shutil.copy(oldpath, newpath)
+    return newpath
     
 
 class LibraryTableButtons(QWidget):

win_venv/tests.py

@@ -1 +1,26 @@
-#tests
+import unittest
+from main import *
+
+# test saving file
+class savingFileTest(unittest.TestCase):
+
+    def test_text_file(self):
+        print('Testing save_input method \n')
+        old_file_content = 'testing\n\n\nmore testing'
+        with open('test.txt', 'w') as file:
+            file.write(old_file_content)
+        
+        script_path = os.path.dirname(os.path.realpath(__file__))
+        test_file_path = script_path + '\\test.txt'
+        newpath = save_input(test_file_path)
+        
+        new_file_content = ''
+        with open(newpath, 'r') as new_file:
+            for line in new_file:
+                new_file_content += line
+        
+        self.assertEqual(old_file_content, new_file_content)
+            
+    
+if __name__ == '__main__':
+    unittest.main()