Skip to content

Commit

Permalink
fix test compile
Browse files Browse the repository at this point in the history
  • Loading branch information
alamb committed Jul 24, 2023
1 parent 063d0c9 commit 05086a3
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 9 deletions.
6 changes: 3 additions & 3 deletions benchmarks/src/options.rs
Original file line number Diff line number Diff line change
Expand Up @@ -24,15 +24,15 @@ use structopt::StructOpt;
pub struct CommonOpt {
/// Number of iterations of each test run
#[structopt(short = "i", long = "iterations", default_value = "3")]
iterations: usize,
pub iterations: usize,

/// Number of partitions to process in parallel
#[structopt(short = "n", long = "partitions", default_value = "2")]
partitions: usize,
pub partitions: usize,

/// Batch size when reading CSV or Parquet files
#[structopt(short = "s", long = "batch-size", default_value = "8192")]
batch_size: usize,
pub batch_size: usize,
}

impl CommonOpt {
Expand Down
18 changes: 12 additions & 6 deletions benchmarks/src/tpch/run.rs
Original file line number Diff line number Diff line change
Expand Up @@ -327,12 +327,15 @@ mod tests {
async fn round_trip_logical_plan(query: usize) -> Result<()> {
let ctx = SessionContext::default();
let path = get_tpch_data_path()?;
let opt = RunOpt {
query: Some(query),
debug: false,
let common = CommonOpt {
iterations: 1,
partitions: 2,
batch_size: 8192,
};
let opt = RunOpt {
query: Some(query),
debug: false,
common,
path: PathBuf::from(path.to_string()),
file_format: "tbl".to_string(),
mem_table: false,
Expand All @@ -356,12 +359,15 @@ mod tests {
async fn round_trip_physical_plan(query: usize) -> Result<()> {
let ctx = SessionContext::default();
let path = get_tpch_data_path()?;
let opt = RunOpt {
query: Some(query),
debug: false,
let common = CommonOpt {
iterations: 1,
partitions: 2,
batch_size: 8192,
};
let opt = RunOpt {
query: Some(query),
debug: false,
common,
path: PathBuf::from(path.to_string()),
file_format: "tbl".to_string(),
mem_table: false,
Expand Down

0 comments on commit 05086a3

Please sign in to comment.