Add tests

nolze · Sep 11, 2018 · 3c5cd5e · 3c5cd5e
1 parent b16cade
commit 3c5cd5e
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/msoffcrypto/format/xls97.py b/msoffcrypto/format/xls97.py
@@ -587,6 +587,18 @@ def decrypt(self, ofile):
         return
 
     def is_encrypted(self):
+        r'''
+        Test if the file is encrypted.
+
+            >>> f = open("tests/inputs/plain.xls", "rb")
+            >>> file = Xls97File(f)
+            >>> file.is_encrypted()
+            False
+            >>> f = open("tests/inputs/rc4cryptoapi_password.xls", "rb")
+            >>> file = Xls97File(f)
+            >>> file.is_encrypted()
+            True
+        '''
         # Utilising the method above, check for encryption type.
         self.data.workbook.seek(0)
         workbook = _BIFFStream(self.data.workbook)

diff --git a/tests/inputs/plain.xls b/tests/inputs/plain.xls