| 305 | } |
| 306 | |
| 307 | void CheckIndices(const NCB::TTrainingDataProvider& dataProvider, |
| 308 | const TFeatureParallelDataSetsHolder& dataSet) { |
| 309 | for (ui32 i = 0; i < dataSet.PermutationsCount(); ++i) { |
| 310 | auto permutation = GetPermutation(dataProvider, i); |
| 311 | TVector<ui32> order; |
| 312 | permutation.FillOrder(order); |
| 313 | TVector<ui32> inverseOrder; |
| 314 | permutation.FillInversePermutation(inverseOrder); |
| 315 | |
| 316 | auto& ds = dataSet.GetDataSetForPermutation(i); |
| 317 | { |
| 318 | TVector<ui32> tmp; |
| 319 | ds.GetIndices().Read(tmp); |
| 320 | for (ui32 i = 0; i < order.size(); ++i) { |
| 321 | UNIT_ASSERT_VALUES_EQUAL(order[i], tmp[i]); |
| 322 | } |
| 323 | ds.GetInverseIndices().Read(tmp); |
| 324 | for (ui32 i = 0; i < order.size(); ++i) { |
| 325 | UNIT_ASSERT_VALUES_EQUAL(inverseOrder[i], tmp[i]); |
| 326 | } |
| 327 | } |
| 328 | } |
| 329 | } |
| 330 | |
| 331 | void TestDatasetHolderBuilder(ui32 binarization, |
| 332 | ui32 permutationCount, |
no test coverage detected