diff tests/test_sort.py @ 135:12649a88545c

stubbing: hooking this up to command line
author Jeff Hammel <k0scist@gmail.com>
date Tue, 17 Mar 2015 11:53:33 -0700
parents
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/tests/test_sort.py	Tue Mar 17 11:53:33 2015 -0700
@@ -0,0 +1,44 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+unit tests for sort
+"""
+
+import unittest
+from numerics.sort import Sorter
+
+class SortUnitTest(unittest.TestCase):
+
+    def test_sort(self):
+        """test array sorter"""
+
+        data = [[1,2,3],
+                [5,6,4],
+                [2,7,2]]
+
+        # forward
+        self.assertEqual(Sorter((-1, True))(data),
+                         [[2,7,2],
+                          [1,2,3],
+                          [5,6,4]])
+        self.assertEqual(Sorter((0, True))(data),
+                         [[1,2,3],
+                          [2,7,2],
+                          [5,6,4]])
+        self.assertEqual(Sorter((1, True))(data),
+                         [[1,2,3],
+                          [5,6,4],
+                          [2,7,2]])
+    def test_reverse(self):
+        data = [[1,2,3],
+                [5,6,4],
+                [2,7,2]]
+        self.assertEqual(Sorter((-1, False))(data),
+                         [[5,6,4],
+                          [1,2,3],
+                          [2,7,2]])
+
+if __name__ == '__main__':
+    unittest.main()
+