Merge pull request #67 from larissalages/tests/radix

add unit tests for radix python

Author: larissalages

classical_algorithms/python/radix.py

@@ -21,6 +21,7 @@ def _sort(self, data):
 				data.extend(bucket)
 		return data
 
+# TODO: Expand functionality to work with negative numbers
 
 # Ex:
 array = [50, 2, 13, 23, 11, 100]

classical_algorithms/python/tests/test_radix.py

@@ -0,0 +1,32 @@
+import unittest
+from classical_algorithms.python.radix import Radix
+
+
+class TestRadix(unittest.TestCase):
+    def test_radix(self):
+        radix = Radix()
+
+        print('None Input')
+        self.assertRaises(TypeError, radix.sort, None)
+
+        print('Empty Input')
+        self.assertEqual(radix.sort([]), [])
+
+        print('One Element')
+        self.assertEqual(radix.sort([25]), [25])
+
+        print('Two or More Elements')
+        array = [10, 1, 0, 100, 5, 15, 100, 7, 500, 200]
+        self.assertEqual(sorted(array), radix.sort(array))
+
+        #  TODO: Include tests for negative numbers
+
+        print('Success: test_radix\n')
+
+
+def main():
+    TestRadix.test_radix()
+
+
+if __name__ == '__main__':
+    main()

readme.md

@@ -16,7 +16,7 @@ Code | Test
 ------------ | -------------
 [Quick Sort](https://github.com/larissalages/code_problems/blob/master/classical_algorithms/python/quicksort.py) | [Unit Test](https://github.com/larissalages/code_problems/blob/master/classical_algorithms/python/tests/test_quick_sort.py)
 [Merge Sort](https://github.com/larissalages/code_problems/blob/master/classical_algorithms/python/mergesort.py) | [Unit Test](https://github.com/larissalages/code_problems/blob/master/classical_algorithms/python/tests/test_merge_sort.py)
-[Radix](https://github.com/larissalages/code_problems/blob/master/classical_algorithms/python/radix.py) | Missing tests
+[Radix](https://github.com/larissalages/code_problems/blob/master/classical_algorithms/python/radix.py) | [Unit Test](https://github.com/larissalages/code_problems/blob/f0353fac71e3b773dbfb12080c01b02baa35ac72/classical_algorithms/python/tests/test_radix.py)
 [Selection Sort](https://github.com/larissalages/code_problems/blob/master/classical_algorithms/python/selection%20sort.py) | Missing tests
 [Binary Search](https://github.com/larissalages/code_problems/blob/master/classical_algorithms/python/Binary%20search.py) | Missing tests