Mercurial > hg > numerics
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() +