/
test_version.cc
60 lines (47 loc) · 1.39 KB
/
test_version.cc
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
/*!
* Copyright 2019 XGBoost contributors
*/
#include <dmlc/io.h>
#include <gtest/gtest.h>
#include <xgboost/base.h>
#include <xgboost/json.h>
#include <xgboost/version_config.h>
#include <string>
#include "../../../src/common/version.h"
#include "../filesystem.h" // dmlc::TemporaryDirectory
namespace xgboost {
TEST(Version, Basic) {
Json j_ver { Object() };
Version::Save(&j_ver);
auto triplet { Version::Load(j_ver) };
ASSERT_TRUE(Version::Same(triplet));
dmlc::TemporaryDirectory tempdir;
const std::string fname = tempdir.path + "/version";
{
std::unique_ptr<dmlc::Stream> fo(dmlc::Stream::Create(fname.c_str(), "w"));
Version::Save(fo.get());
}
{
std::unique_ptr<dmlc::Stream> fi(dmlc::Stream::Create(fname.c_str(), "r"));
auto triplet { Version::Load(fi.get())};;
ASSERT_TRUE(Version::Same(triplet));
}
std::string str { Version::String(triplet) };
size_t ptr {0};
XGBoostVersionT v {0};
v = std::stoi(str, &ptr);
ASSERT_EQ(str.at(ptr), '.');
ASSERT_EQ(v, XGBOOST_VER_MAJOR) << "major: " << v;
str = str.substr(ptr+1);
ptr = 0;
v = std::stoi(str, &ptr);
ASSERT_EQ(str.at(ptr), '.');
ASSERT_EQ(v, XGBOOST_VER_MINOR) << "minor: " << v;;
str = str.substr(ptr+1);
ptr = 0;
v = std::stoi(str, &ptr);
ASSERT_EQ(v, XGBOOST_VER_PATCH) << "patch: " << v;;
str = str.substr(ptr);
ASSERT_EQ(str.size(), 0);
}
} // namespace xgboost