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
79
80
|
#include "catch.hpp"
#include "common/fastopt.h"
#include "3rd_party/yaml-cpp/yaml.h"
using namespace marian;
TEST_CASE("FastOpt can be constructed from a YAML node", "[fastopt]") {
SECTION("from a simple node") {
YAML::Node node = YAML::Load("{foo: bar}");
const FastOpt o(node);
CHECK( o.has("foo") );
CHECK_FALSE( o.has("bar") );
CHECK_FALSE( o.has("baz") );
}
SECTION("from a sequence node") {
YAML::Node node = YAML::Load("{foo: [bar, baz]}");
const FastOpt o(node);
CHECK( o.has("foo") );
}
SECTION("from nested nodes") {
YAML::Node node = YAML::Load("{foo: {bar: 123, baz}}");
const FastOpt o(node);
CHECK( o.has("foo") );
CHECK( o["foo"].has("bar") );
CHECK( o["foo"].has("baz") );
CHECK( o["foo"]["bar"].as<int>() == 123 );
CHECK( o["foo"]["baz"].isNull() );
}
}
TEST_CASE("Options can be accessed", "[fastopt]") {
YAML::Node node = YAML::Load("{"
"foo: bar,"
"seq: [1, 2, 3],"
"subnode: {"
" baz: [ 111.5, False ],"
" qux: 222,"
" preprocess1: n,"
" preprocess2: d,"
" preprocess3: y,"
" }"
"}");
const FastOpt o(node);
SECTION("using operator[]") {
auto& oo = o["subnode"];
CHECK( oo.has("baz") );
CHECK( oo.has("qux") );
CHECK_NOTHROW( o["subnode"]["baz"] );
}
SECTION("using as<T>()") {
CHECK( o["foo"].as<std::string>() == "bar" );
CHECK( o["subnode"]["baz"][0].as<float>() == 111.5f );
CHECK( o["subnode"]["baz"][1].as<bool>() == false );
CHECK( o["subnode"]["baz"][0].as<int>() == 111 );
CHECK( o["subnode"]["preprocess1"].as<std::string>() == "n" ); // don't allow "n" to be cast to boolean false while converting from YAML
CHECK( o["subnode"]["preprocess2"].as<std::string>() == "d" );
CHECK( o["subnode"]["preprocess3"].as<std::string>() == "y" ); // don't allow "y" to be cast to boolean true while converting from YAML
}
node["foo"] = "baz";
if(o.has("foo")) {
FastOpt temp(node["foo"]);
const_cast<FastOpt&>(o["foo"]).swap(temp);
}
CHECK( o["foo"].as<std::string>() == "baz" );
// for(auto k : o[subnode].keys())
// o[subnode][k].type()
SECTION("using as<std::vector<T>>()") {
CHECK( o["seq"].as<std::vector<double>>() == std::vector<double>({1, 2, 3}) );
}
}
|
