summaryrefslogtreecommitdiff
path: root/tools/swig/test/testArray.py
diff options
context:
space:
mode:
authorMathieu Lamarre <mlamarre@ea.com>2020-03-13 13:12:04 -0400
committerMathieu Lamarre <mlamarre@ea.com>2020-03-13 13:12:04 -0400
commit8aad05e63d3efba35f24ace2061b68188fe79b45 (patch)
tree770f1ee8e3fd118e92253cc539a1309e16bcabc7 /tools/swig/test/testArray.py
parent33351f646e0b523139fbba4986eba54a3bfe55a7 (diff)
downloadnumpy-8aad05e63d3efba35f24ace2061b68188fe79b45.tar.gz
Fix swig tests
resize typemap issue python 3 bytes vs str issue
Diffstat (limited to 'tools/swig/test/testArray.py')
-rwxr-xr-xtools/swig/test/testArray.py92
1 files changed, 46 insertions, 46 deletions
diff --git a/tools/swig/test/testArray.py b/tools/swig/test/testArray.py
index 627257622..49011bb13 100755
--- a/tools/swig/test/testArray.py
+++ b/tools/swig/test/testArray.py
@@ -24,24 +24,24 @@ class Array1TestCase(unittest.TestCase):
def testConstructor0(self):
"Test Array1 default constructor"
a = Array.Array1()
- self.failUnless(isinstance(a, Array.Array1))
- self.failUnless(len(a) == 0)
+ self.assertTrue(isinstance(a, Array.Array1))
+ self.assertTrue(len(a) == 0)
def testConstructor1(self):
"Test Array1 length constructor"
- self.failUnless(isinstance(self.array1, Array.Array1))
+ self.assertTrue(isinstance(self.array1, Array.Array1))
def testConstructor2(self):
"Test Array1 array constructor"
na = np.arange(self.length)
aa = Array.Array1(na)
- self.failUnless(isinstance(aa, Array.Array1))
+ self.assertTrue(isinstance(aa, Array.Array1))
def testConstructor3(self):
"Test Array1 copy constructor"
for i in range(self.array1.length()): self.array1[i] = i
arrayCopy = Array.Array1(self.array1)
- self.failUnless(arrayCopy == self.array1)
+ self.assertTrue(arrayCopy == self.array1)
def testConstructorBad(self):
"Test Array1 length constructor, negative"
@@ -49,23 +49,23 @@ class Array1TestCase(unittest.TestCase):
def testLength(self):
"Test Array1 length method"
- self.failUnless(self.array1.length() == self.length)
+ self.assertTrue(self.array1.length() == self.length)
def testLen(self):
"Test Array1 __len__ method"
- self.failUnless(len(self.array1) == self.length)
+ self.assertTrue(len(self.array1) == self.length)
def testResize0(self):
"Test Array1 resize method, length"
newLen = 2 * self.length
self.array1.resize(newLen)
- self.failUnless(len(self.array1) == newLen)
+ self.assertTrue(len(self.array1) == newLen)
def testResize1(self):
"Test Array1 resize method, array"
a = np.zeros((2*self.length,), dtype='l')
self.array1.resize(a)
- self.failUnless(len(self.array1) == a.size)
+ self.assertTrue(len(self.array1) == a.size)
def testResizeBad(self):
"Test Array1 resize method, negative length"
@@ -77,7 +77,7 @@ class Array1TestCase(unittest.TestCase):
for i in range(n):
self.array1[i] = i*i
for i in range(n):
- self.failUnless(self.array1[i] == i*i)
+ self.assertTrue(self.array1[i] == i*i)
def testSetBad1(self):
"Test Array1 __setitem__ method, negative index"
@@ -98,20 +98,20 @@ class Array1TestCase(unittest.TestCase):
def testAsString(self):
"Test Array1 asString method"
for i in range(self.array1.length()): self.array1[i] = i+1
- self.failUnless(self.array1.asString() == "[ 1, 2, 3, 4, 5 ]")
+ self.assertTrue(self.array1.asString() == "[ 1, 2, 3, 4, 5 ]")
def testStr(self):
"Test Array1 __str__ method"
for i in range(self.array1.length()): self.array1[i] = i-2
- self.failUnless(str(self.array1) == "[ -2, -1, 0, 1, 2 ]")
+ self.assertTrue(str(self.array1) == "[ -2, -1, 0, 1, 2 ]")
def testView(self):
"Test Array1 view method"
for i in range(self.array1.length()): self.array1[i] = i+1
a = self.array1.view()
- self.failUnless(isinstance(a, np.ndarray))
- self.failUnless(len(a) == self.length)
- self.failUnless((a == [1, 2, 3, 4, 5]).all())
+ self.assertTrue(isinstance(a, np.ndarray))
+ self.assertTrue(len(a) == self.length)
+ self.assertTrue((a == [1, 2, 3, 4, 5]).all())
######################################################################
@@ -125,18 +125,18 @@ class Array2TestCase(unittest.TestCase):
def testConstructor0(self):
"Test Array2 default constructor"
a = Array.Array2()
- self.failUnless(isinstance(a, Array.Array2))
- self.failUnless(len(a) == 0)
+ self.assertTrue(isinstance(a, Array.Array2))
+ self.assertTrue(len(a) == 0)
def testConstructor1(self):
"Test Array2 nrows, ncols constructor"
- self.failUnless(isinstance(self.array2, Array.Array2))
+ self.assertTrue(isinstance(self.array2, Array.Array2))
def testConstructor2(self):
"Test Array2 array constructor"
na = np.zeros((3, 4), dtype="l")
aa = Array.Array2(na)
- self.failUnless(isinstance(aa, Array.Array2))
+ self.assertTrue(isinstance(aa, Array.Array2))
def testConstructor3(self):
"Test Array2 copy constructor"
@@ -144,7 +144,7 @@ class Array2TestCase(unittest.TestCase):
for j in range(self.ncols):
self.array2[i][j] = i * j
arrayCopy = Array.Array2(self.array2)
- self.failUnless(arrayCopy == self.array2)
+ self.assertTrue(arrayCopy == self.array2)
def testConstructorBad1(self):
"Test Array2 nrows, ncols constructor, negative nrows"
@@ -156,28 +156,28 @@ class Array2TestCase(unittest.TestCase):
def testNrows(self):
"Test Array2 nrows method"
- self.failUnless(self.array2.nrows() == self.nrows)
+ self.assertTrue(self.array2.nrows() == self.nrows)
def testNcols(self):
"Test Array2 ncols method"
- self.failUnless(self.array2.ncols() == self.ncols)
+ self.assertTrue(self.array2.ncols() == self.ncols)
def testLen(self):
"Test Array2 __len__ method"
- self.failUnless(len(self.array2) == self.nrows*self.ncols)
+ self.assertTrue(len(self.array2) == self.nrows*self.ncols)
def testResize0(self):
"Test Array2 resize method, size"
newRows = 2 * self.nrows
newCols = 2 * self.ncols
self.array2.resize(newRows, newCols)
- self.failUnless(len(self.array2) == newRows * newCols)
+ self.assertTrue(len(self.array2) == newRows * newCols)
def testResize1(self):
"Test Array2 resize method, array"
a = np.zeros((2*self.nrows, 2*self.ncols), dtype='l')
self.array2.resize(a)
- self.failUnless(len(self.array2) == a.size)
+ self.assertTrue(len(self.array2) == a.size)
def testResizeBad1(self):
"Test Array2 resize method, negative nrows"
@@ -198,7 +198,7 @@ class Array2TestCase(unittest.TestCase):
for i in range(m):
self.array2[i] = array1[i]
for i in range(m):
- self.failUnless(self.array2[i] == array1[i])
+ self.assertTrue(self.array2[i] == array1[i])
def testSetGet2(self):
"Test Array2 chained __setitem__, __getitem__ methods"
@@ -209,7 +209,7 @@ class Array2TestCase(unittest.TestCase):
self.array2[i][j] = i*j
for i in range(m):
for j in range(n):
- self.failUnless(self.array2[i][j] == i*j)
+ self.assertTrue(self.array2[i][j] == i*j)
def testSetBad1(self):
"Test Array2 __setitem__ method, negative index"
@@ -241,7 +241,7 @@ class Array2TestCase(unittest.TestCase):
for i in range(self.nrows):
for j in range(self.ncols):
self.array2[i][j] = i+j
- self.failUnless(self.array2.asString() == result)
+ self.assertTrue(self.array2.asString() == result)
def testStr(self):
"Test Array2 __str__ method"
@@ -255,13 +255,13 @@ class Array2TestCase(unittest.TestCase):
for i in range(self.nrows):
for j in range(self.ncols):
self.array2[i][j] = i-j
- self.failUnless(str(self.array2) == result)
+ self.assertTrue(str(self.array2) == result)
def testView(self):
"Test Array2 view method"
a = self.array2.view()
- self.failUnless(isinstance(a, np.ndarray))
- self.failUnless(len(a) == self.nrows)
+ self.assertTrue(isinstance(a, np.ndarray))
+ self.assertTrue(len(a) == self.nrows)
######################################################################
@@ -274,24 +274,24 @@ class ArrayZTestCase(unittest.TestCase):
def testConstructor0(self):
"Test ArrayZ default constructor"
a = Array.ArrayZ()
- self.failUnless(isinstance(a, Array.ArrayZ))
- self.failUnless(len(a) == 0)
+ self.assertTrue(isinstance(a, Array.ArrayZ))
+ self.assertTrue(len(a) == 0)
def testConstructor1(self):
"Test ArrayZ length constructor"
- self.failUnless(isinstance(self.array3, Array.ArrayZ))
+ self.assertTrue(isinstance(self.array3, Array.ArrayZ))
def testConstructor2(self):
"Test ArrayZ array constructor"
na = np.arange(self.length, dtype=np.complex128)
aa = Array.ArrayZ(na)
- self.failUnless(isinstance(aa, Array.ArrayZ))
+ self.assertTrue(isinstance(aa, Array.ArrayZ))
def testConstructor3(self):
"Test ArrayZ copy constructor"
for i in range(self.array3.length()): self.array3[i] = complex(i,-i)
arrayCopy = Array.ArrayZ(self.array3)
- self.failUnless(arrayCopy == self.array3)
+ self.assertTrue(arrayCopy == self.array3)
def testConstructorBad(self):
"Test ArrayZ length constructor, negative"
@@ -299,23 +299,23 @@ class ArrayZTestCase(unittest.TestCase):
def testLength(self):
"Test ArrayZ length method"
- self.failUnless(self.array3.length() == self.length)
+ self.assertTrue(self.array3.length() == self.length)
def testLen(self):
"Test ArrayZ __len__ method"
- self.failUnless(len(self.array3) == self.length)
+ self.assertTrue(len(self.array3) == self.length)
def testResize0(self):
"Test ArrayZ resize method, length"
newLen = 2 * self.length
self.array3.resize(newLen)
- self.failUnless(len(self.array3) == newLen)
+ self.assertTrue(len(self.array3) == newLen)
def testResize1(self):
"Test ArrayZ resize method, array"
a = np.zeros((2*self.length,), dtype=np.complex128)
self.array3.resize(a)
- self.failUnless(len(self.array3) == a.size)
+ self.assertTrue(len(self.array3) == a.size)
def testResizeBad(self):
"Test ArrayZ resize method, negative length"
@@ -327,7 +327,7 @@ class ArrayZTestCase(unittest.TestCase):
for i in range(n):
self.array3[i] = i*i
for i in range(n):
- self.failUnless(self.array3[i] == i*i)
+ self.assertTrue(self.array3[i] == i*i)
def testSetBad1(self):
"Test ArrayZ __setitem__ method, negative index"
@@ -348,20 +348,20 @@ class ArrayZTestCase(unittest.TestCase):
def testAsString(self):
"Test ArrayZ asString method"
for i in range(self.array3.length()): self.array3[i] = complex(i+1,-i-1)
- self.failUnless(self.array3.asString() == "[ (1,-1), (2,-2), (3,-3), (4,-4), (5,-5) ]")
+ self.assertTrue(self.array3.asString() == "[ (1,-1), (2,-2), (3,-3), (4,-4), (5,-5) ]")
def testStr(self):
"Test ArrayZ __str__ method"
for i in range(self.array3.length()): self.array3[i] = complex(i-2,(i-2)*2)
- self.failUnless(str(self.array3) == "[ (-2,-4), (-1,-2), (0,0), (1,2), (2,4) ]")
+ self.assertTrue(str(self.array3) == "[ (-2,-4), (-1,-2), (0,0), (1,2), (2,4) ]")
def testView(self):
"Test ArrayZ view method"
for i in range(self.array3.length()): self.array3[i] = complex(i+1,i+2)
a = self.array3.view()
- self.failUnless(isinstance(a, np.ndarray))
- self.failUnless(len(a) == self.length)
- self.failUnless((a == [1+2j, 2+3j, 3+4j, 4+5j, 5+6j]).all())
+ self.assertTrue(isinstance(a, np.ndarray))
+ self.assertTrue(len(a) == self.length)
+ self.assertTrue((a == [1+2j, 2+3j, 3+4j, 4+5j, 5+6j]).all())
######################################################################