* Added tester for array functions.

master
Wirawan Purwanto 10 years ago
parent 32b3580687
commit 773177d2b0
  1. 55
      TESTS/test_array.py

@ -0,0 +1,55 @@
import numpy
def test_array_hstack1():
# 20150520
from numpy import array
from wpylib.array_tools import array_hstack
print("test_array_hstack1::")
A1 = array([[1.5, 2.3, 1.7],
[3.5, 1.0, 9.7],
[4.5, 3.0, 0.5],
[5.0, 8.1, 2.4]])
A2 = array([12.1, 21.5, 47.1, 32.1])
A3 = array([1.1, 2.5, 7.7, 6.6])
A123 = array_hstack((A1, A2, A3))
A123_refs = array([[ 1.5, 2.3, 1.7, 12.1, 1.1],
[ 3.5, 1. , 9.7, 21.5, 2.5],
[ 4.5, 3. , 0.5, 47.1, 7.7],
[ 5. , 8.1, 2.4, 32.1, 6.6]])
print(A123)
assert numpy.all(A123 == A123_refs)
def test_array_vstack1():
# 20150520
from numpy import array
from wpylib.array_tools import array_vstack
print("test_array_vstack1::")
A1 = array([[1.5, 2.3, 1.7],
[3.5, 1.0, 9.7],
[4.5, 3.0, 0.5],
[5.0, 8.1, 2.4]])
A2 = array([12.1, 21.5, 47.1])
A3 = array([1.1, 2.5, 7.7])
A123 = array_vstack((A1, A2, A3))
A123_refs = array([[ 1.5, 2.3, 1.7],
[ 3.5, 1. , 9.7],
[ 4.5, 3. , 0.5],
[ 5. , 8.1, 2.4],
[ 12.1, 21.5, 47.1],
[ 1.1, 2.5, 7.7]])
print(A123)
assert numpy.all(A123 == A123_refs)
if __name__ == '__main__':
test_array_vstack1()
Loading…
Cancel
Save