|
6 | 6 | #include <vector>
|
7 | 7 |
|
8 | 8 | TEST_F(AgradRev, to_soa_sparse_matrix_matrix_double) {
|
| 9 | + using stan::math::to_soa_sparse_matrix; |
9 | 10 | using stan::math::var;
|
10 | 11 | using stan::math::var_value;
|
11 |
| - using stan::math::to_soa_sparse_matrix; |
12 | 12 | std::vector<int> v{0, 1, 2, 0, 1};
|
13 | 13 | std::vector<int> u{0, 1, 2, 3, 4, 5};
|
14 | 14 | Eigen::VectorXd w(5);
|
15 | 15 | int m = 5;
|
16 | 16 | int n = 5;
|
17 | 17 | w << 1, 2, 3, 4, 5;
|
18 |
| - var_value<Eigen::SparseMatrix<double, Eigen::RowMajor>> w_mat_arena = |
19 |
| - to_soa_sparse_matrix<Eigen::RowMajor>(m, n, w, u, v); |
| 18 | + var_value<Eigen::SparseMatrix<double, Eigen::RowMajor>> w_mat_arena |
| 19 | + = to_soa_sparse_matrix<Eigen::RowMajor>(m, n, w, u, v); |
20 | 20 | for (int i = 0; i < 5; ++i) {
|
21 | 21 | EXPECT_EQ(w_mat_arena.val().valuePtr()[i], w(i));
|
22 | 22 | }
|
23 | 23 | }
|
24 | 24 |
|
25 |
| - |
26 | 25 | TEST_F(AgradRev, to_soa_sparse_matrix_matrix_var) {
|
| 26 | + using stan::math::to_soa_sparse_matrix; |
27 | 27 | using stan::math::var;
|
28 | 28 | using stan::math::var_value;
|
29 |
| - using stan::math::to_soa_sparse_matrix; |
30 | 29 | std::vector<int> v{0, 1, 2, 0, 1};
|
31 | 30 | std::vector<int> u{0, 1, 2, 3, 4, 5};
|
32 | 31 | Eigen::VectorXd w(5);
|
33 | 32 | int m = 5;
|
34 | 33 | int n = 5;
|
35 | 34 | w << 1, 2, 3, 4, 5;
|
36 | 35 | Eigen::Matrix<var, -1, 1> w_var(w);
|
37 |
| - var_value<Eigen::SparseMatrix<double, Eigen::RowMajor>> w_mat_arena = |
38 |
| - to_soa_sparse_matrix<Eigen::RowMajor>(m, n, w_var, u, v); |
| 36 | + var_value<Eigen::SparseMatrix<double, Eigen::RowMajor>> w_mat_arena |
| 37 | + = to_soa_sparse_matrix<Eigen::RowMajor>(m, n, w_var, u, v); |
39 | 38 | for (int i = 0; i < 5; ++i) {
|
40 | 39 | EXPECT_EQ(w_mat_arena.val().valuePtr()[i], w.val()(i));
|
41 | 40 | }
|
42 | 41 | }
|
43 | 42 |
|
44 |
| - |
45 | 43 | TEST_F(AgradRev, to_soa_sparse_matrix_var_matrix) {
|
46 | 44 | using stan::math::to_soa_sparse_matrix;
|
47 | 45 | using stan::math::var;
|
|
0 commit comments