---

test/test_generator.py

@@ -0,0 +1,20 @@
+import sys
+import unittest
+
+sys.path.append('../../')
+from generate import generate
+from functions import primality_test
+
+
+class TestGenerator(unittest.TestCase):
+
+  def test_generate(self):
+    (n, e, d) = generate()
+    self.assertIsInstance(n, int, 'n is correctly returned as an integer.')
+    self.assertIsInstance(e, int, 'e is correctly returned as an integer.')
+    self.assertIsInstance(d, int, 'd is correctly returned as an integer.')
+    self.assertFalse(primality_test(n), 'n is not a prime number.')
+
+
+if __name__ == '__main__':
+  unittest.main()