| 128 | |
| 129 | #[tokio::test] |
| 130 | async fn read_limit() -> Result<()> { |
| 131 | let session_ctx = SessionContext::new(); |
| 132 | let state = session_ctx.state(); |
| 133 | let task_ctx = state.task_ctx(); |
| 134 | let projection = None; |
| 135 | let exec = get_exec(&state, projection, Some(1)).await?; |
| 136 | let batches = collect(exec, task_ctx).await?; |
| 137 | assert_eq!(1, batches.len()); |
| 138 | assert_eq!(4, batches[0].num_columns()); |
| 139 | assert_eq!(1, batches[0].num_rows()); |
| 140 | |
| 141 | Ok(()) |
| 142 | } |
| 143 | |
| 144 | #[tokio::test] |
| 145 | async fn infer_schema() -> Result<()> { |