devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

moved pruning code to SparseVector.h

Browse Source
This commit is contained in:
Erik Schultheis
2021-11-15 22:16:01 +00:00
committed by Rasmus Munk Larsen
parent da79095923
commit 13954c4440
3 changed files with 54 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
test/sparse_vector.cpp
View File

@@ -111,7 +111,7 @@ template<typename Scalar,typename StorageIndex> void sparse_vector(int rows, int
// check copy to dense vector with transpose
refV3.resize(0);
VERIFY_IS_APPROX(refV3 = v1.transpose(),v1.toDense());
VERIFY_IS_APPROX(DenseVector(v1),v1.toDense());
VERIFY_IS_APPROX(DenseVector(v1),v1.toDense());
// test conservative resize
{
@@ -144,6 +144,31 @@ template<typename Scalar,typename StorageIndex> void sparse_vector(int rows, int
}
}
void test_pruning() {
using SparseVectorType = SparseVector<double, 0, int>;
SparseVectorType vec;
auto init_vec = [&](){;
vec.resize(10);
vec.insert(3) = 0.1;
vec.insert(5) = 1.0;
vec.insert(8) = -0.1;
vec.insert(9) = -0.2;
};
init_vec();
VERIFY_IS_EQUAL(vec.nonZeros(), 4);
VERIFY_IS_EQUAL(vec.prune(0.1, 1.0), 2);
VERIFY_IS_EQUAL(vec.nonZeros(), 2);
VERIFY_IS_EQUAL(vec.coeff(5), 1.0);
VERIFY_IS_EQUAL(vec.coeff(9), -0.2);
init_vec();
VERIFY_IS_EQUAL(vec.prune([](double v) { return v >= 0; }), 2);
VERIFY_IS_EQUAL(vec.nonZeros(), 2);
VERIFY_IS_EQUAL(vec.coeff(3), 0.1);
VERIFY_IS_EQUAL(vec.coeff(5), 1.0);
}
EIGEN_DECLARE_TEST(sparse_vector)
{
@@ -159,5 +184,7 @@ EIGEN_DECLARE_TEST(sparse_vector)
CALL_SUBTEST_1(( sparse_vector<double,long int>(r, c) ));
CALL_SUBTEST_1(( sparse_vector<double,short>(r, c) ));
}
CALL_SUBTEST_1(test_pruning());
}