Added unittest for rsa.common.inverse

This unittest tests both execution branches of the function, reducing
randomness of code coverage.

1 files changed, 11 insertions, 1 deletions

diff --git a/tests/test_common.py b/tests/test_common.py
index 453dcc8..ef32f61 100644
--- a/tests/test_common.py
+++ b/tests/test_common.py
@@ -18,7 +18,7 @@
 import unittest
 import struct
 from rsa._compat import byte, b
-from rsa.common import byte_size, bit_size, _bit_size
+from rsa.common import byte_size, bit_size, _bit_size, inverse
 
 
 class TestByte(unittest.TestCase):
@@ -75,3 +75,13 @@ class TestBitSize(unittest.TestCase):
         self.assertEqual(_bit_size(1 << 1024), 1025)
         self.assertEqual(_bit_size((1 << 1024) + 1), 1025)
         self.assertEqual(_bit_size((1 << 1024) - 1), 1024)
+
+
+class TestInverse(unittest.TestCase):
+    def test_normal(self):
+        self.assertEqual(3, inverse(7, 4))
+        self.assertEqual(9, inverse(5, 11))
+
+    def test_not_relprime(self):
+        self.assertRaises(ValueError, inverse, 4, 8)
+        self.assertRaises(ValueError, inverse, 25, 5)