Function TEST

tools/test/TestMatch.cc:357–429

#endif

  TEST(TestMatch, columnSelectionTest) {
    ReaderOptions readerOpts;
    RowReaderOptions rowReaderOpts;
    std::list<uint64_t> includes;
    for (uint64_t i = 0; i < 9; i += 2) {
      includes.push_back(i);
    }
    rowReaderOpts.include(includes);
    std::string filename = findExample("demo-11-none.orc");
    std::unique_ptr<Reader> reader =
        createReader(readLocalFile(filename, readerOpts.getReaderMetrics()), readerOpts);
    std::unique_ptr<RowReader> rowReader = reader->createRowReader(rowReaderOpts);

    EXPECT_EQ(CompressionKind_NONE, reader->getCompression());
    EXPECT_EQ(256 * 1024, reader->getCompressionSize());
    EXPECT_EQ(385, reader->getNumberOfStripes());
    EXPECT_EQ(1920800, reader->getNumberOfRows());
    EXPECT_EQ(10000, reader->getRowIndexStride());
    EXPECT_EQ(5069718, reader->getContentLength());
    EXPECT_EQ(filename, reader->getStreamName());
    EXPECT_THAT(reader->getMetadataKeys(), testing::IsEmpty());
    EXPECT_FALSE(reader->hasMetadataValue("foo"));
    EXPECT_EQ(18446744073709551615UL, rowReader->getRowNumber());

    const Type& rootType = reader->getType();
    EXPECT_EQ(0, rootType.getColumnId());
    EXPECT_EQ(STRUCT, rootType.getKind());
    ASSERT_EQ(9, rootType.getSubtypeCount());
    EXPECT_EQ("_col0", rootType.getFieldName(0));
    EXPECT_EQ("_col1", rootType.getFieldName(1));
    EXPECT_EQ("_col2", rootType.getFieldName(2));
    EXPECT_EQ("_col3", rootType.getFieldName(3));
    EXPECT_EQ("_col4", rootType.getFieldName(4));
    EXPECT_EQ("_col5", rootType.getFieldName(5));
    EXPECT_EQ("_col6", rootType.getFieldName(6));
    EXPECT_EQ("_col7", rootType.getFieldName(7));
    EXPECT_EQ("_col8", rootType.getFieldName(8));
    EXPECT_EQ(INT, rootType.getSubtype(0)->getKind());
    EXPECT_EQ(STRING, rootType.getSubtype(1)->getKind());
    EXPECT_EQ(STRING, rootType.getSubtype(2)->getKind());
    EXPECT_EQ(STRING, rootType.getSubtype(3)->getKind());
    EXPECT_EQ(INT, rootType.getSubtype(4)->getKind());
    EXPECT_EQ(STRING, rootType.getSubtype(5)->getKind());
    EXPECT_EQ(INT, rootType.getSubtype(6)->getKind());
    EXPECT_EQ(INT, rootType.getSubtype(7)->getKind());
    EXPECT_EQ(INT, rootType.getSubtype(8)->getKind());
    for (unsigned int i = 0; i < 9; ++i) {
      EXPECT_EQ(i + 1, rootType.getSubtype(i)->getColumnId()) << "fail on " << i;
    }

    const std::vector<bool> selected = rowReader->getSelectedColumns();
    EXPECT_EQ(true, selected[0]) << "fail on " << 0;
    for (size_t i = 1; i < 10; ++i) {
      EXPECT_EQ(i % 2 == 1 ? true : false, selected[i]) << "fail on " << i;
    }

    unsigned long rowCount = 0;
    std::unique_ptr<ColumnVectorBatch> batch = rowReader->createRowBatch(1024);