1 files changed, 381 insertions, 0 deletions
diff --git a/tools/swig/test/testVector.py b/tools/swig/test/testVector.py
new file mode 100755
index 000000000..e7d019cf7
--- /dev/null
+++ b/tools/swig/test/testVector.py
@@ -0,0 +1,381 @@
+#! /usr/bin/env python
+from __future__ import division, absolute_import, print_function
+
+# System imports
+from   distutils.util import get_platform
+import os
+import sys
+import unittest
+
+# Import NumPy
+import numpy as np
+major, minor = [ int(d) for d in np.__version__.split(".")[:2] ]
+if major == 0: BadListError = TypeError
+else:          BadListError = ValueError
+
+import Vector
+
+######################################################################
+
+class VectorTestCase(unittest.TestCase):
+
+    def __init__(self, methodName="runTest"):
+        unittest.TestCase.__init__(self, methodName)
+        self.typeStr  = "double"
+        self.typeCode = "d"
+
+    # Test the (type IN_ARRAY1[ANY]) typemap
+    def testLength(self):
+        "Test length function"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        length = Vector.__dict__[self.typeStr + "Length"]
+        self.assertEquals(length([5, 12, 0]), 13)
+
+    # Test the (type IN_ARRAY1[ANY]) typemap
+    def testLengthBadList(self):
+        "Test length function with bad list"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        length = Vector.__dict__[self.typeStr + "Length"]
+        self.assertRaises(BadListError, length, [5, "twelve", 0])
+
+    # Test the (type IN_ARRAY1[ANY]) typemap
+    def testLengthWrongSize(self):
+        "Test length function with wrong size"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        length = Vector.__dict__[self.typeStr + "Length"]
+        self.assertRaises(TypeError, length, [5, 12])
+
+    # Test the (type IN_ARRAY1[ANY]) typemap
+    def testLengthWrongDim(self):
+        "Test length function with wrong dimensions"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        length = Vector.__dict__[self.typeStr + "Length"]
+        self.assertRaises(TypeError, length, [[1, 2], [3, 4]])
+
+    # Test the (type IN_ARRAY1[ANY]) typemap
+    def testLengthNonContainer(self):
+        "Test length function with non-container"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        length = Vector.__dict__[self.typeStr + "Length"]
+        self.assertRaises(TypeError, length, None)
+
+    # Test the (type* IN_ARRAY1, int DIM1) typemap
+    def testProd(self):
+        "Test prod function"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        prod = Vector.__dict__[self.typeStr + "Prod"]
+        self.assertEquals(prod([1, 2, 3, 4]), 24)
+
+    # Test the (type* IN_ARRAY1, int DIM1) typemap
+    def testProdBadList(self):
+        "Test prod function with bad list"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        prod = Vector.__dict__[self.typeStr + "Prod"]
+        self.assertRaises(BadListError, prod, [[1, "two"], ["e", "pi"]])
+
+    # Test the (type* IN_ARRAY1, int DIM1) typemap
+    def testProdWrongDim(self):
+        "Test prod function with wrong dimensions"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        prod = Vector.__dict__[self.typeStr + "Prod"]
+        self.assertRaises(TypeError, prod, [[1, 2], [8, 9]])
+
+    # Test the (type* IN_ARRAY1, int DIM1) typemap
+    def testProdNonContainer(self):
+        "Test prod function with non-container"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        prod = Vector.__dict__[self.typeStr + "Prod"]
+        self.assertRaises(TypeError, prod, None)
+
+    # Test the (int DIM1, type* IN_ARRAY1) typemap
+    def testSum(self):
+        "Test sum function"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        sum = Vector.__dict__[self.typeStr + "Sum"]
+        self.assertEquals(sum([5, 6, 7, 8]), 26)
+
+    # Test the (int DIM1, type* IN_ARRAY1) typemap
+    def testSumBadList(self):
+        "Test sum function with bad list"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        sum = Vector.__dict__[self.typeStr + "Sum"]
+        self.assertRaises(BadListError, sum, [3, 4, 5, "pi"])
+
+    # Test the (int DIM1, type* IN_ARRAY1) typemap
+    def testSumWrongDim(self):
+        "Test sum function with wrong dimensions"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        sum = Vector.__dict__[self.typeStr + "Sum"]
+        self.assertRaises(TypeError, sum, [[3, 4], [5, 6]])
+
+    # Test the (int DIM1, type* IN_ARRAY1) typemap
+    def testSumNonContainer(self):
+        "Test sum function with non-container"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        sum = Vector.__dict__[self.typeStr + "Sum"]
+        self.assertRaises(TypeError, sum, True)
+
+    # Test the (type INPLACE_ARRAY1[ANY]) typemap
+    def testReverse(self):
+        "Test reverse function"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        reverse = Vector.__dict__[self.typeStr + "Reverse"]
+        vector = np.array([1, 2, 4], self.typeCode)
+        reverse(vector)
+        self.assertEquals((vector == [4, 2, 1]).all(), True)
+
+    # Test the (type INPLACE_ARRAY1[ANY]) typemap
+    def testReverseWrongDim(self):
+        "Test reverse function with wrong dimensions"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        reverse = Vector.__dict__[self.typeStr + "Reverse"]
+        vector = np.array([[1, 2], [3, 4]], self.typeCode)
+        self.assertRaises(TypeError, reverse, vector)
+
+    # Test the (type INPLACE_ARRAY1[ANY]) typemap
+    def testReverseWrongSize(self):
+        "Test reverse function with wrong size"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        reverse = Vector.__dict__[self.typeStr + "Reverse"]
+        vector = np.array([9, 8, 7, 6, 5, 4], self.typeCode)
+        self.assertRaises(TypeError, reverse, vector)
+
+    # Test the (type INPLACE_ARRAY1[ANY]) typemap
+    def testReverseWrongType(self):
+        "Test reverse function with wrong type"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        reverse = Vector.__dict__[self.typeStr + "Reverse"]
+        vector = np.array([1, 2, 4], 'c')
+        self.assertRaises(TypeError, reverse, vector)
+
+    # Test the (type INPLACE_ARRAY1[ANY]) typemap
+    def testReverseNonArray(self):
+        "Test reverse function with non-array"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        reverse = Vector.__dict__[self.typeStr + "Reverse"]
+        self.assertRaises(TypeError, reverse, [2, 4, 6])
+
+    # Test the (type* INPLACE_ARRAY1, int DIM1) typemap
+    def testOnes(self):
+        "Test ones function"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        ones = Vector.__dict__[self.typeStr + "Ones"]
+        vector = np.zeros(5, self.typeCode)
+        ones(vector)
+        np.testing.assert_array_equal(vector, np.array([1, 1, 1, 1, 1]))
+
+    # Test the (type* INPLACE_ARRAY1, int DIM1) typemap
+    def testOnesWrongDim(self):
+        "Test ones function with wrong dimensions"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        ones = Vector.__dict__[self.typeStr + "Ones"]
+        vector = np.zeros((5, 5), self.typeCode)
+        self.assertRaises(TypeError, ones, vector)
+
+    # Test the (type* INPLACE_ARRAY1, int DIM1) typemap
+    def testOnesWrongType(self):
+        "Test ones function with wrong type"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        ones = Vector.__dict__[self.typeStr + "Ones"]
+        vector = np.zeros((5, 5), 'c')
+        self.assertRaises(TypeError, ones, vector)
+
+    # Test the (type* INPLACE_ARRAY1, int DIM1) typemap
+    def testOnesNonArray(self):
+        "Test ones function with non-array"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        ones = Vector.__dict__[self.typeStr + "Ones"]
+        self.assertRaises(TypeError, ones, [2, 4, 6, 8])
+
+    # Test the (int DIM1, type* INPLACE_ARRAY1) typemap
+    def testZeros(self):
+        "Test zeros function"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        zeros = Vector.__dict__[self.typeStr + "Zeros"]
+        vector = np.ones(5, self.typeCode)
+        zeros(vector)
+        np.testing.assert_array_equal(vector, np.array([0, 0, 0, 0, 0]))
+
+    # Test the (int DIM1, type* INPLACE_ARRAY1) typemap
+    def testZerosWrongDim(self):
+        "Test zeros function with wrong dimensions"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        zeros = Vector.__dict__[self.typeStr + "Zeros"]
+        vector = np.ones((5, 5), self.typeCode)
+        self.assertRaises(TypeError, zeros, vector)
+
+    # Test the (int DIM1, type* INPLACE_ARRAY1) typemap
+    def testZerosWrongType(self):
+        "Test zeros function with wrong type"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        zeros = Vector.__dict__[self.typeStr + "Zeros"]
+        vector = np.ones(6, 'c')
+        self.assertRaises(TypeError, zeros, vector)
+
+    # Test the (int DIM1, type* INPLACE_ARRAY1) typemap
+    def testZerosNonArray(self):
+        "Test zeros function with non-array"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        zeros = Vector.__dict__[self.typeStr + "Zeros"]
+        self.assertRaises(TypeError, zeros, [1, 3, 5, 7, 9])
+
+    # Test the (type ARGOUT_ARRAY1[ANY]) typemap
+    def testEOSplit(self):
+        "Test eoSplit function"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        eoSplit = Vector.__dict__[self.typeStr + "EOSplit"]
+        even, odd = eoSplit([1, 2, 3])
+        self.assertEquals((even == [1, 0, 3]).all(), True)
+        self.assertEquals((odd  == [0, 2, 0]).all(), True)
+
+    # Test the (type* ARGOUT_ARRAY1, int DIM1) typemap
+    def testTwos(self):
+        "Test twos function"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        twos = Vector.__dict__[self.typeStr + "Twos"]
+        vector = twos(5)
+        self.assertEquals((vector == [2, 2, 2, 2, 2]).all(), True)
+
+    # Test the (type* ARGOUT_ARRAY1, int DIM1) typemap
+    def testTwosNonInt(self):
+        "Test twos function with non-integer dimension"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        twos = Vector.__dict__[self.typeStr + "Twos"]
+        self.assertRaises(TypeError, twos, 5.0)
+
+    # Test the (int DIM1, type* ARGOUT_ARRAY1) typemap
+    def testThrees(self):
+        "Test threes function"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        threes = Vector.__dict__[self.typeStr + "Threes"]
+        vector = threes(6)
+        self.assertEquals((vector == [3, 3, 3, 3, 3, 3]).all(), True)
+
+    # Test the (type* ARGOUT_ARRAY1, int DIM1) typemap
+    def testThreesNonInt(self):
+        "Test threes function with non-integer dimension"
+        print(self.typeStr, "... ", end=' ', file=sys.stderr)
+        threes = Vector.__dict__[self.typeStr + "Threes"]
+        self.assertRaises(TypeError, threes, "threes")
+
+######################################################################
+
+class scharTestCase(VectorTestCase):
+    def __init__(self, methodName="runTest"):
+        VectorTestCase.__init__(self, methodName)
+        self.typeStr  = "schar"
+        self.typeCode = "b"
+
+######################################################################
+
+class ucharTestCase(VectorTestCase):
+    def __init__(self, methodName="runTest"):
+        VectorTestCase.__init__(self, methodName)
+        self.typeStr  = "uchar"
+        self.typeCode = "B"
+
+######################################################################
+
+class shortTestCase(VectorTestCase):
+    def __init__(self, methodName="runTest"):
+        VectorTestCase.__init__(self, methodName)
+        self.typeStr  = "short"
+        self.typeCode = "h"
+
+######################################################################
+
+class ushortTestCase(VectorTestCase):
+    def __init__(self, methodName="runTest"):
+        VectorTestCase.__init__(self, methodName)
+        self.typeStr  = "ushort"
+        self.typeCode = "H"
+
+######################################################################
+
+class intTestCase(VectorTestCase):
+    def __init__(self, methodName="runTest"):
+        VectorTestCase.__init__(self, methodName)
+        self.typeStr  = "int"
+        self.typeCode = "i"
+
+######################################################################
+
+class uintTestCase(VectorTestCase):
+    def __init__(self, methodName="runTest"):
+        VectorTestCase.__init__(self, methodName)
+        self.typeStr  = "uint"
+        self.typeCode = "I"
+
+######################################################################
+
+class longTestCase(VectorTestCase):
+    def __init__(self, methodName="runTest"):
+        VectorTestCase.__init__(self, methodName)
+        self.typeStr  = "long"
+        self.typeCode = "l"
+
+######################################################################
+
+class ulongTestCase(VectorTestCase):
+    def __init__(self, methodName="runTest"):
+        VectorTestCase.__init__(self, methodName)
+        self.typeStr  = "ulong"
+        self.typeCode = "L"
+
+######################################################################
+
+class longLongTestCase(VectorTestCase):
+    def __init__(self, methodName="runTest"):
+        VectorTestCase.__init__(self, methodName)
+        self.typeStr  = "longLong"
+        self.typeCode = "q"
+
+######################################################################
+
+class ulongLongTestCase(VectorTestCase):
+    def __init__(self, methodName="runTest"):
+        VectorTestCase.__init__(self, methodName)
+        self.typeStr  = "ulongLong"
+        self.typeCode = "Q"
+
+######################################################################
+
+class floatTestCase(VectorTestCase):
+    def __init__(self, methodName="runTest"):
+        VectorTestCase.__init__(self, methodName)
+        self.typeStr  = "float"
+        self.typeCode = "f"
+
+######################################################################
+
+class doubleTestCase(VectorTestCase):
+    def __init__(self, methodName="runTest"):
+        VectorTestCase.__init__(self, methodName)
+        self.typeStr  = "double"
+        self.typeCode = "d"
+
+######################################################################
+
+if __name__ == "__main__":
+
+    # Build the test suite
+    suite = unittest.TestSuite()
+    suite.addTest(unittest.makeSuite(    scharTestCase))
+    suite.addTest(unittest.makeSuite(    ucharTestCase))
+    suite.addTest(unittest.makeSuite(    shortTestCase))
+    suite.addTest(unittest.makeSuite(   ushortTestCase))
+    suite.addTest(unittest.makeSuite(      intTestCase))
+    suite.addTest(unittest.makeSuite(     uintTestCase))
+    suite.addTest(unittest.makeSuite(     longTestCase))
+    suite.addTest(unittest.makeSuite(    ulongTestCase))
+    suite.addTest(unittest.makeSuite( longLongTestCase))
+    suite.addTest(unittest.makeSuite(ulongLongTestCase))
+    suite.addTest(unittest.makeSuite(    floatTestCase))
+    suite.addTest(unittest.makeSuite(   doubleTestCase))
+
+    # Execute the test suite
+    print("Testing 1D Functions of Module Vector")
+    print("NumPy version", np.__version__)
+    print()
+    result = unittest.TextTestRunner(verbosity=2).run(suite)
+    sys.exit(len(result.errors) + len(result.failures))