Function TEST

dali/core/dev_buffer_test.cc:20–60

namespace dali {

TEST(DeviceBuffer, SizeChanges) {
  DeviceBuffer<int> db;
  db.resize(1);
  EXPECT_EQ(db.size(), 1u);
  EXPECT_GE(db.capacity(), 1);
  db.resize(2);
  EXPECT_EQ(db.size(), 2u);
  EXPECT_GE(db.capacity(), 2u);
  db.resize(3);
  EXPECT_EQ(db.size(), 3u);
  EXPECT_GE(db.capacity(), 4u);
  auto *ptr = db.data();
  db.resize(4);
  EXPECT_EQ(db.size(), 4u);
  EXPECT_GE(db.capacity(), 4u);
  EXPECT_EQ(db.data(), ptr) << "No reallocation should have occurred";
  db.resize(5);
  EXPECT_EQ(db.size(), 5u);
  EXPECT_GE(db.capacity(), 8u);
  ptr = db.data();
  db.resize(8);
  EXPECT_EQ(db.size(), 8u);
  EXPECT_GE(db.capacity(), 8u);
  EXPECT_EQ(db.data(), ptr) << "No reallocation should have occurred";
  db.resize(3);
  EXPECT_EQ(db.size(), 3u);
  EXPECT_GE(db.capacity(), 8u);
  EXPECT_EQ(db.data(), ptr) << "No reallocation should have occurred";
  db.shrink_to_fit();
  EXPECT_GE(db.capacity(), 3u);
  EXPECT_NE(db.data(), ptr) << "Buffer contents must have been copied - no in-place reallocation";
  ptr = db.data();
  db.clear();
  EXPECT_EQ(db.size(), 0u);
  EXPECT_GE(db.capacity(), 3u);
  EXPECT_EQ(db.data(), ptr) << "No reallocation should have occurred";
  db.shrink_to_fit();
  EXPECT_EQ(db.size(), 0u);
  EXPECT_EQ(db.capacity(), 0u);
  EXPECT_EQ(db.data(), nullptr);
}

TEST(DeviceBuffer, FromHost) {
  DeviceBuffer<int> db;