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
|
#include <forstio/test/suite.hpp>
#include "../c++/data.hpp"
#include "../c++/data_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()));
}
}
}
|
