1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
#include <catch2/catch_all.hpp>
#include <random>
#include "sopt/wrapper.h"
TEST_CASE("Function wrappers", "[utility]") {
using namespace sopt;
using t_Array = Array<int>;
using t_RefArray = t_Array&;
using t_ConstRefArray = t_Array const;
SECTION("Square function") {
auto func = [](t_RefArray output, t_ConstRefArray const &input) { output = input * 2 + 1; };
t_Array const x = t_Array::Random(5);
auto const A = details::wrap<t_Array>(func);
// Expected result
t_Array const expected = (x * 2 + 1).eval();
CHECK((A * x).matrix() == expected.matrix());
CHECK(A(x).matrix() == expected.matrix());
}
SECTION("Rectangular function") {
auto func = [](t_RefArray output, t_ConstRefArray const &input) {
output = input.head(input.size() / 2) * 2 + 1;
};
t_Array const x = t_Array::Random(5);
auto const A = details::wrap<t_Array>(func, {{1, 2, 0}});
// Expected result
t_Array const expected = (x.head(x.size() / 2) * 2 + 1).eval();
CHECK((A * x).cols() == 1);
CHECK((A * x).rows() == 2);
CHECK((A * x).matrix() == expected.matrix());
CHECK(A(x).matrix() == expected.matrix());
}
SECTION("Fixed output-size functions") {
auto func = [](t_RefArray output, t_ConstRefArray const &input) {
output = input.head(3) * 2 + 1;
};
t_Array const x = t_Array::Random(5);
auto const A = details::wrap<t_Array>(func, {{0, 1, 3}});
// Expected result
t_Array const expected = (x.head(3) * 2 + 1).eval();
CHECK((A * x).cols() == 1);
CHECK((A * x).rows() == 3);
CHECK((A * x).matrix() == expected.matrix());
CHECK(A(x).matrix() == expected.matrix());
}
}
|
