Span class. (#3548)
* Add basic Span class based on ISO++20. * Use Span<Entry const> instead of Inst in SparsePage. * Add DeviceSpan in HostDeviceVector, use it in regression obj.
This commit is contained in:
trivialfis
committed by
Rory Mitchell
Rory Mitchell
parent
2b7a1c5780
commit
2c502784ff
423
tests/cpp/common/test_span.cc
Normal file
423
tests/cpp/common/test_span.cc
Normal file
@@ -0,0 +1,423 @@
|
||||
/*!
|
||||
* Copyright 2018 XGBoost contributors
|
||||
*/
|
||||
#include <gtest/gtest.h>
|
||||
#include <vector>
|
||||
|
||||
#include "../../../src/common/span.h"
|
||||
#include "test_span.h"
|
||||
|
||||
namespace xgboost {
|
||||
namespace common {
|
||||
|
||||
TEST(Span, TestStatus) {
|
||||
int status = 1;
|
||||
TestTestStatus {&status}();
|
||||
ASSERT_EQ(status, -1);
|
||||
}
|
||||
|
||||
TEST(Span, DlfConstructors) {
|
||||
// Dynamic extent
|
||||
{
|
||||
Span<int> s;
|
||||
ASSERT_EQ(s.size(), 0);
|
||||
ASSERT_EQ(s.data(), nullptr);
|
||||
|
||||
Span<int const> cs;
|
||||
ASSERT_EQ(cs.size(), 0);
|
||||
ASSERT_EQ(cs.data(), nullptr);
|
||||
}
|
||||
|
||||
// Static extent
|
||||
{
|
||||
Span<int, 0> s;
|
||||
ASSERT_EQ(s.size(), 0);
|
||||
ASSERT_EQ(s.data(), nullptr);
|
||||
|
||||
Span<int const, 0> cs;
|
||||
ASSERT_EQ(cs.size(), 0);
|
||||
ASSERT_EQ(cs.data(), nullptr);
|
||||
}
|
||||
|
||||
// Init list.
|
||||
{
|
||||
Span<float> s {};
|
||||
ASSERT_EQ(s.size(), 0);
|
||||
ASSERT_EQ(s.data(), nullptr);
|
||||
|
||||
Span<int const> cs {};
|
||||
ASSERT_EQ(cs.size(), 0);
|
||||
ASSERT_EQ(cs.data(), nullptr);
|
||||
}
|
||||
}
|
||||
|
||||
TEST(Span, FromNullPtr) {
|
||||
// dynamic extent
|
||||
{
|
||||
Span<float> s {nullptr, static_cast<Span<float>::index_type>(0)};
|
||||
ASSERT_EQ(s.size(), 0);
|
||||
ASSERT_EQ(s.data(), nullptr);
|
||||
|
||||
Span<float const> cs {nullptr, static_cast<Span<float>::index_type>(0)};
|
||||
ASSERT_EQ(cs.size(), 0);
|
||||
ASSERT_EQ(cs.data(), nullptr);
|
||||
}
|
||||
// static extent
|
||||
{
|
||||
Span<float, 0> s {nullptr, static_cast<Span<float>::index_type>(0)};
|
||||
ASSERT_EQ(s.size(), 0);
|
||||
ASSERT_EQ(s.data(), nullptr);
|
||||
|
||||
Span<float const, 0> cs {nullptr, static_cast<Span<float>::index_type>(0)};
|
||||
ASSERT_EQ(cs.size(), 0);
|
||||
ASSERT_EQ(cs.data(), nullptr);
|
||||
}
|
||||
}
|
||||
|
||||
TEST(Span, FromPtrLen) {
|
||||
float arr[16];
|
||||
InitializeRange(arr, arr+16);
|
||||
|
||||
// static extent
|
||||
{
|
||||
Span<float> s (arr, 16);
|
||||
ASSERT_EQ (s.size(), 16);
|
||||
ASSERT_EQ (s.data(), arr);
|
||||
|
||||
for (Span<float>::index_type i = 0; i < 16; ++i) {
|
||||
ASSERT_EQ (s[i], arr[i]);
|
||||
}
|
||||
|
||||
Span<float const> cs (arr, 16);
|
||||
ASSERT_EQ (cs.size(), 16);
|
||||
ASSERT_EQ (cs.data(), arr);
|
||||
|
||||
for (Span<float const>::index_type i = 0; i < 16; ++i) {
|
||||
ASSERT_EQ (cs[i], arr[i]);
|
||||
}
|
||||
}
|
||||
|
||||
{
|
||||
EXPECT_ANY_THROW(Span<float> tmp (arr, -1););
|
||||
}
|
||||
|
||||
// dynamic extent
|
||||
{
|
||||
Span<float, 16> s (arr, 16);
|
||||
ASSERT_EQ (s.size(), 16);
|
||||
ASSERT_EQ (s.data(), arr);
|
||||
|
||||
for (size_t i = 0; i < 16; ++i) {
|
||||
ASSERT_EQ (s[i], arr[i]);
|
||||
}
|
||||
|
||||
Span<float const, 16> cs (arr, 16);
|
||||
ASSERT_EQ (cs.size(), 16);
|
||||
ASSERT_EQ (cs.data(), arr);
|
||||
|
||||
for (Span<float const>::index_type i = 0; i < 16; ++i) {
|
||||
ASSERT_EQ (cs[i], arr[i]);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
TEST(Span, FromFirstLast) {
|
||||
float arr[16];
|
||||
InitializeRange(arr, arr+16);
|
||||
|
||||
// dynamic extent
|
||||
{
|
||||
Span<float> s (arr, arr + 16);
|
||||
ASSERT_EQ (s.size(), 16);
|
||||
ASSERT_EQ (s.data(), arr);
|
||||
ASSERT_EQ (s.data() + s.size(), arr + 16);
|
||||
|
||||
for (size_t i = 0; i < 16; ++i) {
|
||||
ASSERT_EQ (s[i], arr[i]);
|
||||
}
|
||||
|
||||
Span<float const> cs (arr, arr + 16);
|
||||
ASSERT_EQ (cs.size(), 16);
|
||||
ASSERT_EQ (cs.data(), arr);
|
||||
ASSERT_EQ (cs.data() + cs.size(), arr + 16);
|
||||
|
||||
for (size_t i = 0; i < 16; ++i) {
|
||||
ASSERT_EQ (cs[i], arr[i]);
|
||||
}
|
||||
}
|
||||
|
||||
// static extent
|
||||
{
|
||||
Span<float, 16> s (arr, arr + 16);
|
||||
ASSERT_EQ (s.size(), 16);
|
||||
ASSERT_EQ (s.data(), arr);
|
||||
ASSERT_EQ (s.data() + s.size(), arr + 16);
|
||||
|
||||
for (size_t i = 0; i < 16; ++i) {
|
||||
ASSERT_EQ (s[i], arr[i]);
|
||||
}
|
||||
|
||||
Span<float const> cs (arr, arr + 16);
|
||||
ASSERT_EQ (cs.size(), 16);
|
||||
ASSERT_EQ (cs.data(), arr);
|
||||
ASSERT_EQ (cs.data() + cs.size(), arr + 16);
|
||||
|
||||
for (size_t i = 0; i < 16; ++i) {
|
||||
ASSERT_EQ (cs[i], arr[i]);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
struct BaseClass {
|
||||
virtual void operator()() {}
|
||||
};
|
||||
struct DerivedClass : public BaseClass {
|
||||
virtual void operator()() {}
|
||||
};
|
||||
|
||||
TEST(Span, FromOther) {
|
||||
|
||||
// convert constructor
|
||||
{
|
||||
Span<DerivedClass> derived;
|
||||
Span<BaseClass> base { derived };
|
||||
ASSERT_EQ(base.size(), derived.size());
|
||||
ASSERT_EQ(base.data(), derived.data());
|
||||
}
|
||||
|
||||
float arr[16];
|
||||
InitializeRange(arr, arr + 16);
|
||||
|
||||
// default copy constructor
|
||||
{
|
||||
Span<float> s0 (arr);
|
||||
Span<float> s1 (s0);
|
||||
ASSERT_EQ(s0.size(), s1.size());
|
||||
ASSERT_EQ(s0.data(), s1.data());
|
||||
}
|
||||
}
|
||||
|
||||
TEST(Span, FromArray) {
|
||||
float arr[16];
|
||||
InitializeRange(arr, arr + 16);
|
||||
|
||||
{
|
||||
Span<float> s (arr);
|
||||
ASSERT_EQ(&arr[0], s.data());
|
||||
ASSERT_EQ(s.size(), 16);
|
||||
for (size_t i = 0; i < 16; ++i) {
|
||||
ASSERT_EQ(arr[i], s[i]);
|
||||
}
|
||||
}
|
||||
|
||||
{
|
||||
Span<float, 16> s (arr);
|
||||
ASSERT_EQ(&arr[0], s.data());
|
||||
ASSERT_EQ(s.size(), 16);
|
||||
for (size_t i = 0; i < 16; ++i) {
|
||||
ASSERT_EQ(arr[i], s[i]);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
TEST(Span, FromContainer) {
|
||||
std::vector<float> vec (16);
|
||||
InitializeRange(vec.begin(), vec.end());
|
||||
|
||||
Span<float> s(vec);
|
||||
ASSERT_EQ(s.size(), vec.size());
|
||||
ASSERT_EQ(s.data(), vec.data());
|
||||
|
||||
bool res = std::equal(vec.begin(), vec.end(), s.begin());
|
||||
ASSERT_TRUE(res);
|
||||
}
|
||||
|
||||
TEST(Span, Assignment) {
|
||||
int status = 1;
|
||||
TestAssignment{&status}();
|
||||
ASSERT_EQ(status, 1);
|
||||
}
|
||||
|
||||
TEST(SpanIter, Construct) {
|
||||
int status = 1;
|
||||
TestIterConstruct{&status}();
|
||||
ASSERT_EQ(status, 1);
|
||||
}
|
||||
|
||||
TEST(SpanIter, Ref) {
|
||||
int status = 1;
|
||||
TestIterRef{&status}();
|
||||
ASSERT_EQ(status, 1);
|
||||
}
|
||||
|
||||
TEST(SpanIter, Calculate) {
|
||||
int status = 1;
|
||||
TestIterCalculate{&status}();
|
||||
ASSERT_EQ(status, 1);
|
||||
}
|
||||
|
||||
TEST(SpanIter, Compare) {
|
||||
int status = 1;
|
||||
TestIterCompare{&status}();
|
||||
ASSERT_EQ(status, 1);
|
||||
}
|
||||
|
||||
TEST(Span, BeginEnd) {
|
||||
int status = 1;
|
||||
TestBeginEnd{&status}();
|
||||
ASSERT_EQ(status, 1);
|
||||
}
|
||||
|
||||
TEST(Span, RBeginREnd) {
|
||||
int status = 1;
|
||||
TestRBeginREnd{&status}();
|
||||
ASSERT_EQ(status, 1);
|
||||
}
|
||||
|
||||
TEST(Span, ElementAccess) {
|
||||
float arr[16];
|
||||
InitializeRange(arr, arr + 16);
|
||||
|
||||
Span<float> s (arr);
|
||||
size_t j = 0;
|
||||
for (auto i : s) {
|
||||
ASSERT_EQ(i, arr[j]);
|
||||
++j;
|
||||
}
|
||||
|
||||
EXPECT_ANY_THROW(s[16]);
|
||||
EXPECT_ANY_THROW(s[-1]);
|
||||
|
||||
EXPECT_ANY_THROW(s(16));
|
||||
EXPECT_ANY_THROW(s(-1));
|
||||
}
|
||||
|
||||
TEST(Span, Obversers) {
|
||||
int status = 1;
|
||||
TestObservers{&status}();
|
||||
ASSERT_EQ(status, 1);
|
||||
}
|
||||
|
||||
TEST(Span, FirstLast) {
|
||||
// static extent
|
||||
{
|
||||
float arr[16];
|
||||
InitializeRange(arr, arr + 16);
|
||||
|
||||
Span<float> s (arr);
|
||||
Span<float, 4> first = s.first<4>();
|
||||
|
||||
ASSERT_EQ(first.size(), 4);
|
||||
ASSERT_EQ(first.data(), arr);
|
||||
|
||||
for (size_t i = 0; i < first.size(); ++i) {
|
||||
ASSERT_EQ(first[i], arr[i]);
|
||||
}
|
||||
|
||||
EXPECT_ANY_THROW(s.first<-1>());
|
||||
EXPECT_ANY_THROW(s.first<17>());
|
||||
EXPECT_ANY_THROW(s.first<32>());
|
||||
}
|
||||
|
||||
{
|
||||
float arr[16];
|
||||
InitializeRange(arr, arr + 16);
|
||||
|
||||
Span<float> s (arr);
|
||||
Span<float, 4> last = s.last<4>();
|
||||
|
||||
ASSERT_EQ(last.size(), 4);
|
||||
ASSERT_EQ(last.data(), arr + 12);
|
||||
|
||||
for (size_t i = 0; i < last.size(); ++i) {
|
||||
ASSERT_EQ(last[i], arr[i+12]);
|
||||
}
|
||||
|
||||
EXPECT_ANY_THROW(s.last<-1>());
|
||||
EXPECT_ANY_THROW(s.last<17>());
|
||||
EXPECT_ANY_THROW(s.last<32>());
|
||||
}
|
||||
|
||||
// dynamic extent
|
||||
{
|
||||
float *arr = new float[16];
|
||||
InitializeRange(arr, arr + 16);
|
||||
Span<float> s (arr, 16);
|
||||
Span<float> first = s.first(4);
|
||||
|
||||
ASSERT_EQ(first.size(), 4);
|
||||
ASSERT_EQ(first.data(), s.data());
|
||||
|
||||
for (size_t i = 0; i < first.size(); ++i) {
|
||||
ASSERT_EQ(first[i], s[i]);
|
||||
}
|
||||
|
||||
EXPECT_ANY_THROW(s.first(-1));
|
||||
EXPECT_ANY_THROW(s.first(17));
|
||||
EXPECT_ANY_THROW(s.first(32));
|
||||
|
||||
delete [] arr;
|
||||
}
|
||||
|
||||
{
|
||||
float *arr = new float[16];
|
||||
InitializeRange(arr, arr + 16);
|
||||
Span<float> s (arr, 16);
|
||||
Span<float> last = s.last(4);
|
||||
|
||||
ASSERT_EQ(last.size(), 4);
|
||||
ASSERT_EQ(last.data(), s.data() + 12);
|
||||
|
||||
for (size_t i = 0; i < last.size(); ++i) {
|
||||
ASSERT_EQ(s[12 + i], last[i]);
|
||||
}
|
||||
|
||||
EXPECT_ANY_THROW(s.last(-1));
|
||||
EXPECT_ANY_THROW(s.last(17));
|
||||
EXPECT_ANY_THROW(s.last(32));
|
||||
|
||||
delete [] arr;
|
||||
}
|
||||
}
|
||||
|
||||
TEST(Span, Subspan) {
|
||||
int arr[16] {0};
|
||||
Span<int> s1 (arr);
|
||||
auto s2 = s1.subspan<4>();
|
||||
ASSERT_EQ(s1.size() - 4, s2.size());
|
||||
|
||||
auto s3 = s1.subspan(2, 4);
|
||||
ASSERT_EQ(s1.data() + 2, s3.data());
|
||||
ASSERT_EQ(s3.size(), 4);
|
||||
|
||||
auto s4 = s1.subspan(2, dynamic_extent);
|
||||
ASSERT_EQ(s1.data() + 2, s4.data());
|
||||
ASSERT_EQ(s4.size(), s1.size() - 2);
|
||||
|
||||
EXPECT_ANY_THROW(s1.subspan(-1, 0));
|
||||
EXPECT_ANY_THROW(s1.subspan(16, 0));
|
||||
|
||||
EXPECT_ANY_THROW(s1.subspan<-1>());
|
||||
EXPECT_ANY_THROW(s1.subspan<16>());
|
||||
}
|
||||
|
||||
TEST(Span, Compare) {
|
||||
int status = 1;
|
||||
TestCompare{&status}();
|
||||
ASSERT_EQ(status, 1);
|
||||
}
|
||||
|
||||
TEST(Span, AsBytes) {
|
||||
int status = 1;
|
||||
TestAsBytes{&status}();
|
||||
ASSERT_EQ(status, 1);
|
||||
}
|
||||
|
||||
TEST(Span, AsWritableBytes) {
|
||||
int status = 1;
|
||||
TestAsWritableBytes{&status}();
|
||||
ASSERT_EQ(status, 1);
|
||||
}
|
||||
|
||||
} // namespace common
|
||||
} // namespace xgboost
|
||||
Reference in New Issue
Block a user