Function TestRangePredicates

c++/test/TestPredicatePushdown.cc:71–119

  }

  void TestRangePredicates(Reader* reader) {
    // Build search argument (x >= 300000 AND x < 600000) for column 'int1'.
    // Test twice for using column name and column id respectively.
    for (int k = 0; k < 2; ++k) {
      std::unique_ptr<SearchArgument> sarg;
      if (k == 0) {
        sarg =
            SearchArgumentFactory::newBuilder()
                ->startAnd()
                .startNot()
                .lessThan("int1", PredicateDataType::LONG, Literal(static_cast<int64_t>(300000L)))
                .end()
                .lessThan("int1", PredicateDataType::LONG, Literal(static_cast<int64_t>(600000L)))
                .end()
                .build();
      } else {
        sarg = SearchArgumentFactory::newBuilder()
                   ->startAnd()
                   .startNot()
                   .lessThan(/*columnId=*/1, PredicateDataType::LONG,
                             Literal(static_cast<int64_t>(300000L)))
                   .end()
                   .lessThan(/*columnId=*/1, PredicateDataType::LONG,
                             Literal(static_cast<int64_t>(600000L)))
                   .end()
                   .build();
      }

      RowReaderOptions rowReaderOpts;
      rowReaderOpts.searchArgument(std::move(sarg));
      auto rowReader = reader->createRowReader(rowReaderOpts);

      auto readBatch = rowReader->createRowBatch(2000);
      auto& batch0 = dynamic_cast<StructVectorBatch&>(*readBatch);
      auto& batch1 = dynamic_cast<LongVectorBatch&>(*batch0.fields[0]);
      auto& batch2 = dynamic_cast<StringVectorBatch&>(*batch0.fields[1]);

      EXPECT_EQ(true, rowReader->next(*readBatch));
      EXPECT_EQ(1000, readBatch->numElements);
      EXPECT_EQ(1000, rowReader->getRowNumber());
      for (uint64_t i = 1000; i < 2000; ++i) {
        EXPECT_EQ(300 * i, batch1.data[i - 1000]);
        EXPECT_EQ(std::to_string(10 * i),
                  std::string(batch2.data[i - 1000], static_cast<size_t>(batch2.length[i - 1000])));
      }
      EXPECT_EQ(false, rowReader->next(*readBatch));
      EXPECT_EQ(3500, rowReader->getRowNumber());
    }
  }

  void TestNoRowsSelected(Reader* reader) {
    // Look through the file with no rows selected: x < 0