parent
32b3580687
commit
773177d2b0
1 changed files with 55 additions and 0 deletions
@ -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…
Reference in new issue