Function TestSubsetFeatures

catboost/libs/data/ut/objects_ut.cpp:407–492

    void TestSubsetFeatures(
        TMaybe<TArraySubsetIndexing<ui32>> subsetForGetSubset,
        TMaybe<EObjectsOrder> objectsOrderForGetSubset,
        const TObjectsGrouping& expectedObjectsGrouping,
        const TCommonObjectsData& commonData,

        // SubsetIndexing is not checked, only TRawObjectsDataProvider's fields
        const TCommonObjectsData& expectedCommonData,
        const TVector<TVector<float>>& srcFloatFeatures,
        const TVector<TVector<float>>& subsetFloatFeatures,
        const TVector<THashMap<ui32, TString>>& catFeaturesHashToString,
        const TVector<TVector<ui32>>& srcCatFeatures,
        const TVector<TVector<ui32>>& subsetCatFeatures
    ) {
        NPar::TLocalExecutor localExecutor;
        localExecutor.RunAdditionalThreads(2);

        // (use float, use cat) pairs
        for (auto useFeatureTypes : {
            std::make_pair(true, false),
            std::make_pair(false, true),
            std::make_pair(true, true)
        }) {
            auto objectsDataProvider = GetMaybeSubsetDataProvider(
                CreateRawObjectsDataProvider(
                    srcFloatFeatures,
                    srcCatFeatures,
                    catFeaturesHashToString,
                    commonData,
                    useFeatureTypes,
                    &localExecutor
                ),
                subsetForGetSubset,
                objectsOrderForGetSubset,
                &localExecutor
            );

            ui32 featureCount = 0;
            TVector<ui32> catFeatureIndices;

            UNIT_ASSERT_EQUAL(expectedObjectsGrouping, *objectsDataProvider.GetObjectsGrouping());

            if (useFeatureTypes.first) {
                featureCount += srcFloatFeatures.size();

                for (auto i : xrange(subsetFloatFeatures.size())) {
                    UNIT_ASSERT(
                        Equal<float>(
                            *(*objectsDataProvider.GetFloatFeature(i))->ExtractValues(&localExecutor),
                            subsetFloatFeatures[i]
                        )
                    );
                }
            }
            if (useFeatureTypes.second) {
                catFeatureIndices.resize(srcCatFeatures.size());
                std::iota(catFeatureIndices.begin(), catFeatureIndices.end(), featureCount);