Add unittest based test module.
authorMadhusudan.C.S <madhusudancs@gmail.com>
Fri, 03 Sep 2010 11:51:33 +0530
changeset 119 9f353900cee8
parent 118 513d43e25927
child 120 7428e411bd7a
Add unittest based test module.
tdd/math_utils/test_gcd.py
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/tdd/math_utils/test_gcd.py	Fri Sep 03 11:51:33 2010 +0530
@@ -0,0 +1,29 @@
+import gcd
+import unittest
+
+class TestGcdFunction(unittest.TestCase):
+
+    def setUp(self):
+        self.test_file = open('gcd_testcases.dat')
+        self.test_cases = []
+        for line in self.test_file:
+            values = line.split(', ')
+            a = int(values[0])
+            b = int(values[1])
+            g = int(values[2])
+
+            self.test_cases.append([a, b, g])
+
+    def test_gcd(self):
+        for case in self.test_cases:
+            a = case[0]
+            b = case[1]
+            g = case[2]
+            self.assertEqual(gcd.gcd(a, b), g)
+
+    def tearDown(self):
+        self.test_file.close()
+        del self.test_cases
+
+if __name__ == '__main__':
+    unittest.main()