| 162 | }; |
| 163 | |
| 164 | void CheckBins(const TFeatureParallelDataSet& dataSet, |
| 165 | const TBinarizedFeaturesManager& featuresManager, |
| 166 | const TBinarySplit& split, ui32 depth, |
| 167 | const TCudaBuffer<ui32, NCudaLib::TMirrorMapping>& bins, |
| 168 | TVector<ui32>& currentBins) { |
| 169 | auto& dataProvider = dataSet.GetDataProvider(); |
| 170 | auto& permutation = dataSet.GetCtrsEstimationPermutation(); |
| 171 | |
| 172 | TCpuTreeCtrHelper helper(featuresManager, dataProvider, permutation); |
| 173 | helper.Split(split, depth, currentBins); |
| 174 | TVector<ui32> gpuBins; |
| 175 | bins.Read(gpuBins); |
| 176 | for (ui32 i = 0; i < currentBins.size(); ++i) { |
| 177 | UNIT_ASSERT_VALUES_EQUAL(currentBins[i], gpuBins[i]); |
| 178 | } |
| 179 | } |
| 180 | |
| 181 | void TestTreeBuilder(ui32 binarization, |
| 182 | ui32 permutationCount, |
no test coverage detected