Skip to content

Commit e2ab314

Browse files
committed
test: fix
1 parent 90cf65c commit e2ab314

File tree

2 files changed

+9
-4
lines changed

2 files changed

+9
-4
lines changed

unittest/eigen_ref.cpp

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,8 +42,13 @@ Eigen::Ref<MatType> editBlock(Eigen::Ref<MatType> mat, Eigen::DenseIndex i,
4242
Eigen::DenseIndex j, Eigen::DenseIndex n,
4343
Eigen::DenseIndex m) {
4444
typename Eigen::Ref<MatType>::BlockXpr B = mat.block(i, j, n, m);
45-
Eigen::Map<VectorXd> view(B.data(), B.size());
46-
view.setLinSpaced(0., (double)view.size() - 1.);
45+
int k = 0;
46+
for (int i = 0; i < B.rows(); ++i) {
47+
for (int j = 0; j < B.cols(); ++j) {
48+
B(i, j) = k++;
49+
}
50+
}
51+
std::cout << "B:\n" << B << std::endl;
4752
return mat;
4853
}
4954

unittest/python/test_eigen_ref.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,6 @@ def test(mat):
2222
assert np.all(ref == mat)
2323

2424
const_ref = asConstRef(mat)
25-
# import pdb; pdb.set_trace()
2625
assert np.all(const_ref == mat)
2726

2827
mat.fill(0.0)
@@ -37,11 +36,12 @@ def test(mat):
3736

3837
mat.fill(0.0)
3938
mat_as_C_order = np.array(mat, order="F")
40-
mat_copy = mat_as_C_order.copy()
4139
getBlock(mat_as_C_order, 0, 0, 3, 2)[:, :] = 1.0
4240

4341
assert np.all(mat_as_C_order[:3, :2] == np.ones((3, 2)))
4442

43+
mat_as_C_order[:3, :2] = 0.0
44+
mat_copy = mat_as_C_order.copy()
4545
editBlock(mat_as_C_order, 0, 0, 3, 2)
4646
mat_copy[:3, :2] = np.arange(6).reshape(3, 2)
4747

0 commit comments

Comments
 (0)