| 140 | size_t constexpr CSRIterForTest::kCols; |
| 141 | |
| 142 | int CSRSetDataNextForTest(DataIterHandle data_handle, |
| 143 | XGBCallbackSetData *set_function, |
| 144 | DataHolderHandle set_function_handle) { |
| 145 | size_t constexpr kIters { 2 }; |
| 146 | auto iter = static_cast<CSRIterForTest *>(data_handle); |
| 147 | if (iter->Iter() < kIters) { |
| 148 | auto batch = iter->Next(); |
| 149 | batch.columns = CSRIterForTest::kCols; |
| 150 | set_function(set_function_handle, batch); |
| 151 | return 1; |
| 152 | } else { |
| 153 | return 0; // stoping condition |
| 154 | } |
| 155 | } |
| 156 | |
| 157 | TEST(Adapter, IteratorAdapter) { |
| 158 | CSRIterForTest iter; |