add factorial

tdoly · tdoly · commit 8dbd53bf5a80 · 2014-01-05T22:13:39.000+08:00
diff --git a/Factorial/Python/factorial.py b/Factorial/Python/factorial.py
@@ -0,0 +1,17 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+'''
+File: factorial.py
+Author: tdoly
+Description: http://en.wikipedia.org/wiki/Factorial
+'''
+
+def factorial(num):
+    if num < 0:
+         raise "the input number is error"
+    elif num == 0:
+        return 1
+    else:
+        return reduce((lambda x, y: x*y), [i for i in range(num+1) if i > 0])
+
diff --git a/Factorial/Python/factorial_test.py b/Factorial/Python/factorial_test.py
@@ -0,0 +1,24 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+'''
+File: factorial_test.py
+Author: tdoly
+Description: the factorial test
+'''
+
+import unittest
+from factorial import factorial
+
+class TestFactorial(unittest.TestCase):
+    def template(self, test_list, test_result, method):
+        for num, result in zip(test_list, test_result):
+            self.assertEqual(method(num), result, msg="ERROR!factorial(%s) != %s" % (num, result))
+
+    def test_factorial(self):
+        test_list = (0, 1, 2, 3, 4, 5, 10)
+        test_result = (1, 1, 2, 6, 24, 120, 3628800)
+        self.template(test_list, test_result, factorial)
+
+if __name__ == '__main__':
+    unittest.main()
