Unit tests.

laurentluce · laurentluce · commit dbd22bcdc51d · 2015-11-10T13:39:19.000-05:00
diff --git a/algorithms/a_star_path_finding.py b/algorithms/a_star_path_finding.py
@@ -104,20 +104,6 @@ def get_path(self):
         path.reverse()
         return path
 
-    def compare(self, cell1, cell2):
-        """
-        Compare 2 cells F values
-
-        @param cell1 1st cell
-        @param cell2 2nd cell
-        @returns -1, 0 or 1 if lower, equal or greater
-        """
-        if cell1.f < cell2.f:
-            return -1
-        elif cell1.f > cell2.f:
-            return 1
-        return 0
-
     def update_cell(self, adj, cell):
         """
         Update adjacent cell
diff --git a/algorithms/generators.py b/algorithms/generators.py
@@ -5,9 +5,12 @@ def fib(n):
     Example: for i in fib(5): print i
     @param n fib range upper bound
     """
+    if not n:
+        return
     a, b = 0, 1
+    yield a
     i = 0
-    while i < n:
+    while i < n - 1:
         yield b
         a, b = b, a+b
         i += 1
diff --git a/algorithms/permutations.py b/algorithms/permutations.py
@@ -1,7 +1,7 @@
 def permutations(l):
     """
     Generator for list permutations
-    
+
     @param l list to generate permutations for
     @result yield each permutation
 
@@ -25,7 +25,3 @@ def permutations(l):
         for p in permutations(l):
             for i in range(len(p)+1):
                 yield p[:i] + a + p[i:]
-
-for p in permutations([1,2,3]):
-    print p
-
diff --git a/algorithms/tests/all_tests.py b/algorithms/tests/all_tests.py
@@ -0,0 +1,20 @@
+"""Run all of the tests."""
+import os
+import sys
+import unittest2 as unittest
+
+def main(args=None):
+    unittest_dir = '.'
+    unittest_suite = unittest.defaultTestLoader.discover(unittest_dir)
+
+    kwargs = {}
+    if args and '-v' in args:
+        kwargs['verbosity'] = 2
+    runner = unittest.TextTestRunner(sys.stdout, "Unittests",
+                                     **kwargs)
+    results = runner.run(unittest_suite)
+    return results.wasSuccessful()
+
+if __name__ == '__main__':
+    status = main(sys.argv[1:])
+    sys.exit(int(not status))
diff --git a/algorithms/tests/coverage.sh b/algorithms/tests/coverage.sh
@@ -0,0 +1,2 @@
+coverage run --source=$PYTHONPATH/algorithms --omit=$PYTHONPATH/algorithms/tests/* all_tests.py
+coverage report --omit=$PYTHONPATH/algorithms/tests/* -m
diff --git a/algorithms/tests/test_generators.py b/algorithms/tests/test_generators.py
@@ -0,0 +1,20 @@
+import unittest
+import algorithms.generators as generators
+
+class GeneratorsTest(unittest.TestCase):
+
+    def setUp(self):
+        pass
+
+    def test_fib(self):
+        fib = [e for e in generators.fib(10)]
+        self.assertEqual(fib, [0, 1, 1, 2, 3, 5, 8, 13, 21, 34])
+
+    def test_fib_empty(self):
+        fib = [e for e in generators.fib(0)]
+        self.assertEqual(fib, [])
+
+
+if __name__ == '__main__':
+  unittest.main()
+
diff --git a/algorithms/tests/test_list.py b/algorithms/tests/test_list.py
@@ -0,0 +1,17 @@
+import unittest
+import algorithms.list as list
+
+class List(unittest.TestCase):
+
+    def setUp(self):
+        pass
+
+    def test_list(self):
+        bounds, m = [e for e in list.find_max_sub([-2, 3, -4, 5, 1, -5])]
+        self.assertEqual(bounds, (3, 4))
+        self.assertEqual(m, 6)
+
+
+if __name__ == '__main__':
+  unittest.main()
+
diff --git a/algorithms/tests/test_permutations.py b/algorithms/tests/test_permutations.py
@@ -0,0 +1,21 @@
+import unittest
+import algorithms.permutations as permutations
+
+class GeneratorsTest(unittest.TestCase):
+
+    def setUp(self):
+        pass
+
+    def test_permutations(self):
+        p = [e for e in permutations.permutations([1, 2, 3])]
+        self.assertEqual(p, [[1, 2, 3], [2, 1, 3], [2, 3, 1], [1, 3, 2],
+                             [3, 1, 2], [3, 2, 1]])
+
+    def test_permutations_single(self):
+        p = [e for e in permutations.permutations([1])]
+        self.assertEqual(p, [[1]])
+
+
+if __name__ == '__main__':
+  unittest.main()
+
diff --git a/algorithms/tests/test_string_matching.py b/algorithms/tests/test_string_matching.py
@@ -1,8 +1,8 @@
 import unittest
-import string_matching
+import algorithms.string_matching as string_matching
 
 class StringMatchingTest(unittest.TestCase):
-    
+
   def test_string_matching_naive(self):
         t = 'ababbababa'
         s = 'aba'

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,2 @@`
	`1`	`+coverage run --source=$PYTHONPATH/algorithms --omit=$PYTHONPATH/algorithms/tests/* all_tests.py`
	`2`	`+coverage report --omit=$PYTHONPATH/algorithms/tests/* -m`