From 29d202d869e96e50637f9bb1a96703e169c560f7 Mon Sep 17 00:00:00 2001
From: "Sybren A. St?vel" <sybren@stuvel.eu>
Date: Mon, 18 Jan 2016 15:39:50 +0100
Subject: Updated key loading example and included it as unit test, fixes bug
 #31

---
 tests/test_load_save_keys.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'tests/test_load_save_keys.py')

diff --git a/tests/test_load_save_keys.py b/tests/test_load_save_keys.py
index 64f75de..6d87cf9 100644
--- a/tests/test_load_save_keys.py
+++ b/tests/test_load_save_keys.py
@@ -18,6 +18,8 @@
 
 import base64
 import unittest
+import os.path
+
 from rsa._compat import b
 
 import rsa.key
@@ -141,3 +143,13 @@ class PemTest(unittest.TestCase):
         self.assertEqual(CLEAN_PUBLIC_PEM, pem)
 
 
+    def test_load_from_disk(self):
+        """Test loading a PEM file from disk."""
+
+        fname = os.path.join(os.path.dirname(__file__), 'private.pem')
+        with open(fname, mode='rb') as privatefile:
+            keydata = privatefile.read()
+        privkey = rsa.key.PrivateKey.load_pkcs1(keydata)
+
+        self.assertEqual(15945948582725241569, privkey.p)
+        self.assertEqual(14617195220284816877, privkey.q)
-- 
cgit v1.2.1