File: test_transform_iterator.cc

package info (click to toggle)
xgboost 3.0.0-1
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 13,796 kB
  • sloc: cpp: 67,502; python: 35,503; java: 4,676; ansic: 1,426; sh: 1,320; xml: 1,197; makefile: 204; javascript: 19
file content (20 lines) | stat: -rw-r--r-- 456 bytes parent folder | download | duplicates (2) 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
 
/**
 * Copyright 2022 by XGBoost Contributors
 */
#include <gtest/gtest.h>

#include <cstddef>  // std::size_t

#include "../../../src/common/transform_iterator.h"

namespace xgboost {
namespace common {
TEST(IndexTransformIter, Basic) {
  auto sqr = [](std::size_t i) { return i * i; };
  auto iter = MakeIndexTransformIter(sqr);
  for (std::size_t i = 0; i < 4; ++i) {
    ASSERT_EQ(iter[i], sqr(i));
  }
}
}  // namespace common
}  // namespace xgboost