| 226 | |
| 227 | #[tokio::test] |
| 228 | async fn read_limit() -> Result<()> { |
| 229 | let session_ctx = SessionContext::new(); |
| 230 | let state = session_ctx.state(); |
| 231 | let task_ctx = session_ctx.task_ctx(); |
| 232 | let projection = Some(vec![0, 1, 2, 3]); |
| 233 | let exec = |
| 234 | get_exec(&state, "aggregate_test_100.csv", projection, Some(1), true).await?; |
| 235 | let batches = collect(exec, task_ctx).await?; |
| 236 | assert_eq!(1, batches.len()); |
| 237 | assert_eq!(4, batches[0].num_columns()); |
| 238 | assert_eq!(1, batches[0].num_rows()); |
| 239 | |
| 240 | Ok(()) |
| 241 | } |
| 242 | |
| 243 | #[tokio::test] |
| 244 | async fn infer_schema() -> Result<()> { |