mirror of
https://github.com/moses-smt/mosesdecoder.git
synced 2024-11-14 10:47:21 +03:00
36 lines
785 B
C++
36 lines
785 B
C++
#include "util/stream/io.hh"
|
|
|
|
#include "util/stream/stream.hh"
|
|
#include "util/file.hh"
|
|
|
|
#define BOOST_TEST_MODULE StreamTest
|
|
#include <boost/test/unit_test.hpp>
|
|
|
|
#include <unistd.h>
|
|
|
|
namespace util { namespace stream { namespace {
|
|
|
|
BOOST_AUTO_TEST_CASE(StreamTest) {
|
|
scoped_fd in(MakeTemp("io_test_temp"));
|
|
for (uint64_t i = 0; i < 100000; ++i) {
|
|
WriteOrThrow(in.get(), &i, sizeof(uint64_t));
|
|
}
|
|
SeekOrThrow(in.get(), 0);
|
|
|
|
ChainConfig config;
|
|
config.entry_size = 8;
|
|
config.total_memory = 100;
|
|
config.block_count = 12;
|
|
|
|
Stream s;
|
|
Chain chain(config);
|
|
chain >> Read(in.get()) >> s >> kRecycle;
|
|
uint64_t i = 0;
|
|
for (; s; ++s, ++i) {
|
|
BOOST_CHECK_EQUAL(i, *static_cast<const uint64_t*>(s.Get()));
|
|
}
|
|
BOOST_CHECK_EQUAL(100000ULL, i);
|
|
}
|
|
|
|
}}} // namespaces
|