Zengtudor/pybind11
1
0
Fork 0
mirror of https://github.com/pybind/pybind11.git synced 2024-11-11 08:03:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

Test eigen converts slices of 3d arrays correctly

Browse Source
This commit is contained in:
Ben North 2016-07-05 21:03:19 +01:00
parent 3e0e779322
commit 7b8d9e0246
2 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
example/eigen.py
View File

@ -10,6 +10,7 @@ from example import dense_passthrough_r, dense_passthrough_c
from example import sparse_r, sparse_c from example import sparse_r, sparse_c
from example import sparse_passthrough_r, sparse_passthrough_c from example import sparse_passthrough_r, sparse_passthrough_c
from example import double_row, double_col from example import double_row, double_col
from example import double_mat_cm, double_mat_rm
import numpy as np import numpy as np
ref = np.array( ref = np.array(
@ -55,3 +56,10 @@ print("double_row(first_row) = %s" % check_got_vs_ref(double_row(first_row), 2.0
print("double_col(first_row) = %s" % check_got_vs_ref(double_col(first_row), 2.0 * first_row)) print("double_col(first_row) = %s" % check_got_vs_ref(double_col(first_row), 2.0 * first_row))
print("double_row(first_col) = %s" % check_got_vs_ref(double_row(first_col), 2.0 * first_col)) print("double_row(first_col) = %s" % check_got_vs_ref(double_row(first_col), 2.0 * first_col))
print("double_col(first_col) = %s" % check_got_vs_ref(double_col(first_col), 2.0 * first_col)) print("double_col(first_col) = %s" % check_got_vs_ref(double_col(first_col), 2.0 * first_col))
counting_3d = np.arange(27.0, dtype=np.float32).reshape((3, 3, 3))
slices = [counting_3d[0, :, :], counting_3d[:, 0, :], counting_3d[:, :, 0]]
for slice_idx, ref_mat in enumerate(slices):
print("double_mat_cm(%d) = %s" % (slice_idx, check_got_vs_ref(double_mat_cm(ref_mat), 2.0 * ref_mat)))
print("double_mat_rm(%d) = %s" % (slice_idx, check_got_vs_ref(double_mat_rm(ref_mat), 2.0 * ref_mat)))

6
example/eigen.ref
View File

@ -20,3 +20,9 @@ double_row(first_row) = OK
double_col(first_row) = OK double_col(first_row) = OK
double_row(first_col) = OK double_row(first_col) = OK
double_col(first_col) = OK double_col(first_col) = OK
double_mat_cm(0) = OK
double_mat_rm(0) = OK
double_mat_cm(1) = OK
double_mat_rm(1) = OK
double_mat_cm(2) = OK
double_mat_rm(2) = OK