(store_url: &url::Url)
| 1242 | } |
| 1243 | |
| 1244 | fn build_ctx(store_url: &url::Url) -> Arc<TaskContext> { |
| 1245 | let tmp_dir = tempfile::TempDir::new().unwrap(); |
| 1246 | let local = Arc::new( |
| 1247 | LocalFileSystem::new_with_prefix(&tmp_dir) |
| 1248 | .expect("should create object store"), |
| 1249 | ); |
| 1250 | |
| 1251 | let mut session = SessionConfig::default(); |
| 1252 | let mut parquet_opts = ParquetOptions { |
| 1253 | allow_single_file_parallelism: true, |
| 1254 | ..Default::default() |
| 1255 | }; |
| 1256 | parquet_opts.allow_single_file_parallelism = true; |
| 1257 | session.options_mut().execution.parquet = parquet_opts; |
| 1258 | |
| 1259 | let runtime = RuntimeEnv::default(); |
| 1260 | runtime |
| 1261 | .object_store_registry |
| 1262 | .register_store(store_url, local); |
| 1263 | |
| 1264 | Arc::new( |
| 1265 | TaskContext::default() |
| 1266 | .with_session_config(session) |
| 1267 | .with_runtime(Arc::new(runtime)), |
| 1268 | ) |
| 1269 | } |
| 1270 | |
| 1271 | #[tokio::test] |
| 1272 | async fn parquet_sink_write() -> Result<()> { |
searching dependent graphs…