Add SeekEnd to MemoryFixSizeBuffer. (#109)

* Don't assert buffer size.
This commit is contained in:
Jiaming Yuan
2019-10-13 00:09:25 -04:00
committed by GitHub
parent 5d1b613910
commit 6dab74689c
5 changed files with 34 additions and 7 deletions

View File

@@ -2,7 +2,10 @@ find_package(GTest REQUIRED)
add_executable(
unit_tests
allreduce_base_test.cc allreduce_robust_test.cc allreduce_mock_test.cc
test_io.cc
allreduce_robust_test.cc
allreduce_base_test.cc
allreduce_mock_test.cc
test_main.cpp)
target_link_libraries(

18
test/cpp/test_io.cc Normal file
View File

@@ -0,0 +1,18 @@
/*!
* Copyright (c) 2019 by Contributors
*/
#include <gtest/gtest.h>
#include <rabit/internal/io.h>
#include <vector>
namespace rabit {
TEST(MemoryFixSizeBuffer, Seek) {
size_t constexpr kSize { 64 };
std::vector<int32_t> memory( kSize );
utils::MemoryFixSizeBuffer buf(memory.data(), memory.size());
buf.Seek(utils::MemoryFixSizeBuffer::SeekEnd);
size_t end = buf.Tell();
ASSERT_EQ(end, kSize);
}
} // namespace rabit