Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Implement ubjson. (#7549)

Browse Source 
* Implement ubjson.

This is a partial implementation of UBJSON with support for typed arrays.  Some missing
features are `f64`, typed object, and the no-op.
This commit is contained in:
Jiaming Yuan
2022-01-10 23:24:23 +08:00
committed by GitHub
parent 001503186c
commit c635d4c46a
6 changed files with 792 additions and 242 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
tests/cpp/test_serialization.cc
View File

@@ -1,15 +1,25 @@
// Copyright (c) 2019-2020 by Contributors
// Copyright (c) 2019-2022 by Contributors
#include <gtest/gtest.h>
#include <dmlc/filesystem.h>
#include <string>
#include <xgboost/learner.h>
#include <xgboost/data.h>
#include <xgboost/base.h>
#include <xgboost/json.h>
#include "helpers.h"
#include "../../src/common/io.h"
#include "../../src/common/random.h"
namespace xgboost {
template <typename Array>
void CompareIntArray(Json l, Json r) {
auto const& l_arr = get<Array const>(l);
auto const& r_arr = get<Array const>(r);
ASSERT_EQ(l_arr.size(), r_arr.size());
for (size_t i = 0; i < l_arr.size(); ++i) {
ASSERT_EQ(l_arr[i], r_arr[i]);
}
}
void CompareJSON(Json l, Json r) {
switch (l.GetValue().Type()) {
@@ -45,6 +55,27 @@ void CompareJSON(Json l, Json r) {
}
break;
}
case Value::ValueKind::kNumberArray: {
auto const& l_arr = get<F32Array const>(l);
auto const& r_arr = get<F32Array const>(r);
ASSERT_EQ(l_arr.size(), r_arr.size());
for (size_t i = 0; i < l_arr.size(); ++i) {
ASSERT_NEAR(l_arr[i], r_arr[i], kRtEps);
}
break;
}
case Value::ValueKind::kU8Array: {
CompareIntArray<U8Array>(l, r);
break;
}
case Value::ValueKind::kI32Array: {
CompareIntArray<I32Array>(l, r);
break;
}
case Value::ValueKind::kI64Array: {
CompareIntArray<I64Array>(l, r);
break;
}
case Value::ValueKind::kBoolean: {
ASSERT_EQ(l, r);
break;