| 70 | |
| 71 | #[tokio::test] |
| 72 | async fn read_limit() -> Result<()> { |
| 73 | let session_ctx = SessionContext::new(); |
| 74 | let state = session_ctx.state(); |
| 75 | let task_ctx = state.task_ctx(); |
| 76 | let projection = None; |
| 77 | let exec = get_exec(&state, "alltypes_plain.avro", projection, Some(1)).await?; |
| 78 | let batches = collect(exec, task_ctx).await?; |
| 79 | assert_eq!(1, batches.len()); |
| 80 | assert_eq!(11, batches[0].num_columns()); |
| 81 | assert_eq!(1, batches[0].num_rows()); |
| 82 | |
| 83 | Ok(()) |
| 84 | } |
| 85 | |
| 86 | #[tokio::test] |
| 87 | async fn read_alltypes_plain_avro() -> Result<()> { |