Skip to content

Commit 3ce1089

Browse files
committed
Add plan assertions to tpch tests
1 parent e902e1f commit 3ce1089

File tree

2 files changed

+792
-35
lines changed

2 files changed

+792
-35
lines changed

src/test_utils/tpch.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -160,14 +160,14 @@ where
160160

161161
macro_rules! must_generate_tpch_table {
162162
($generator:ident, $arrow:ident, $name:literal, $data_dir:expr) => {
163-
let data_dir = $data_dir.join(format!("{}.parquet", $name));
163+
let data_dir = $data_dir.join($name);
164164
fs::create_dir_all(data_dir.clone()).expect("Failed to create data directory");
165165
// create three partitions for the table
166166
(1..=3).for_each(|part| {
167167
generate_table(
168168
// TODO: Consider adjusting the partitions and batch sizes.
169169
$arrow::new($generator::new(SCALE_FACTOR, part, 3)).with_batch_size(1000),
170-
&format!("{}.parquet", part),
170+
&format!("{part}"),
171171
&data_dir.clone().into_boxed_path(),
172172
)
173173
.expect(concat!("Failed to generate ", $name, " table"));

0 commit comments

Comments
 (0)