diff --git a/cpp/src/arrow/dataset/scanner_benchmark.cc b/cpp/src/arrow/dataset/scanner_benchmark.cc index be953b35559..287d76418ff 100644 --- a/cpp/src/arrow/dataset/scanner_benchmark.cc +++ b/cpp/src/arrow/dataset/scanner_benchmark.cc @@ -162,7 +162,11 @@ void ScanOnly( acero::DeclarationToTable(std::move(scan))); ASSERT_GT(collected->num_rows(), 0); - ASSERT_EQ(collected->num_columns(), 2); + if (factory_name == "scan") { + ASSERT_EQ(collected->num_columns(), 6); + } else if (factory_name == "scan2") { + ASSERT_EQ(collected->num_columns(), 2); + } } static constexpr int kScanIdx = 0;