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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
#include <forstio/test/suite.hpp>
#include "../c++/data.hpp"
#include "../c++/data_math.hpp"
#include "../c++/math.hpp"
#include "../c++/csv.hpp"
#include <iostream>
namespace {
namespace sch {
using namespace saw::schema;
}
/*
SAW_TEST("Math/Basic"){
using namespace saw;
data<sch::Vector<sch::Float64,2u>> a;
data<sch::Vector<sch::Float64,2u>> b;
auto c = a + b;
}
*/
SAW_TEST("Math/Tensor"){
using namespace saw;
data<sch::Matrix<sch::Float64,2u,2u> > a;
{
a.at({{0u,0u}}) = 1.0;
a.at({{0u,1u}}) = 2.0;
a.at({{1u,0u}}) = 3.0;
a.at({{1u,1u}}) = 4.0;
}
data<sch::Matrix<sch::Float64,2u,2u> > b;
{
b.at({{0u,0u}}) = 2.0;
b.at({{0u,1u}}) = 3.0;
b.at({{1u,0u}}) = 4.0;
b.at({{1u,1u}}) = 5.0;
}
auto c = a + b;
{
SAW_EXPECT(c.at({{0u,0u}}).get() == 3.0, std::string{"Unexpected value at (0,0): "} + std::to_string(c.at({{0u,0u}}).get()) );
SAW_EXPECT(c.at({{0u,1u}}).get() == 5.0, std::string{"Unexpected value at (0,1): "} + std::to_string(c.at({{0u,1u}}).get()));
SAW_EXPECT(c.at({{1u,0u}}).get() == 7.0, std::string{"Unexpected value at (1,0): "} + std::to_string(c.at({{1u,0u}}).get()));
SAW_EXPECT(c.at({{1u,1u}}).get() == 9.0, std::string{"Unexpected value at (1,1): "} + std::to_string(c.at({{1u,1u}}).get()));
}
auto d = b - a;
{
SAW_EXPECT(d.at({{0u,0u}}).get() == 1.0, std::string{"Unexpected value at (0,0): "} + std::to_string(d.at({{0u,0u}}).get()) );
SAW_EXPECT(d.at({{0u,1u}}).get() == 1.0, std::string{"Unexpected value at (0,1): "} + std::to_string(d.at({{0u,1u}}).get()));
SAW_EXPECT(d.at({{1u,0u}}).get() == 1.0, std::string{"Unexpected value at (1,0): "} + std::to_string(d.at({{1u,0u}}).get()));
SAW_EXPECT(d.at({{1u,1u}}).get() == 1.0, std::string{"Unexpected value at (1,1): "} + std::to_string(d.at({{1u,1u}}).get()));
}
}
SAW_TEST("Math/Dot"){
using namespace saw;
data<sch::Vector<sch::Float64, 2u>> a;
{
a.at({{0u}}) = 2.0;
a.at({{1u}}) = 1.0;
}
data<sch::Vector<sch::Float64, 2u>> b;
{
b.at({{0u}}) = -1.0;
b.at({{1u}}) = 5.0;
}
auto c = math::dot(a,b);
SAW_EXPECT(c.at({}).get() == 3.0, std::string{"Unexpected value for dot product "} + std::to_string(c.at({}).get()) );
}
}
|
