summaryrefslogtreecommitdiff
path: root/numpy/doc/swig/test/testVector.py
diff options
context:
space:
mode:
Diffstat (limited to 'numpy/doc/swig/test/testVector.py')
-rwxr-xr-xnumpy/doc/swig/test/testVector.py380
1 files changed, 0 insertions, 380 deletions
diff --git a/numpy/doc/swig/test/testVector.py b/numpy/doc/swig/test/testVector.py
deleted file mode 100755
index 2ee918389..000000000
--- a/numpy/doc/swig/test/testVector.py
+++ /dev/null
@@ -1,380 +0,0 @@
-#! /usr/bin/env python
-
-# 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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 >>sys.stderr, self.typeStr, "... ",
- 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))