feat(test): support more tests

Signed-off-by: Alan Tang <[email protected]>

Author: Standing-Man

datafusion/sqllogictest/test_files/explain_tree.slt

@@ -463,32 +463,6 @@ physical_plan
 15)│          rows: 1          │
 16)└───────────────────────────┘
 
-# Query with projection on memory
-query TT
-explain SELECT int_col, bigint_col, int_col+bigint_col AS sum_col FROM table3;
-----
-logical_plan
-01)Projection: table3.int_col, table3.bigint_col, CAST(table3.int_col AS Int64) + table3.bigint_col AS sum_col
-02)--TableScan: table3 projection=[int_col, bigint_col]
-physical_plan
-01)┌───────────────────────────┐
-02)│       ProjectionExec      │
-03)│    --------------------   │
-04)│   ProjectionExec: expr:   │
-05)│   [int_col@0 as int_col,  │
-06)│       bigint_col@1 as     │
-07)│      bigint_col, CAST     │
-08)│   (int_col@0 AS Int64) +  │
-09)│  bigint_col@1 as sum_col] │
-10)└─────────────┬─────────────┘
-11)┌─────────────┴─────────────┐
-12)│       DataSourceExec      │
-13)│    --------------------   │
-14)│        bytes: 1560        │
-15)│       format: memory      │
-16)│          rows: 1          │
-17)└───────────────────────────┘
-
 # Query with filter on json
 query TT
 explain SELECT int_col FROM table4 WHERE string_col != 'foo';
@@ -545,6 +519,150 @@ physical_plan
 17)│       format: arrow       │
 18)└───────────────────────────┘
 
+
+# Query with projection on csv
+query TT
+explain SELECT int_col, bigint_col, int_col+bigint_col AS sum_col FROM table1;
+----
+logical_plan
+01)Projection: table1.int_col, table1.bigint_col, CAST(table1.int_col AS Int64) + table1.bigint_col AS sum_col
+02)--TableScan: table1 projection=[int_col, bigint_col]
+physical_plan
+01)┌───────────────────────────┐
+02)│       ProjectionExec      │
+03)│    --------------------   │
+04)│   ProjectionExec: expr:   │
+05)│   [int_col@0 as int_col,  │
+06)│       bigint_col@1 as     │
+07)│      bigint_col, CAST     │
+08)│   (int_col@0 AS Int64) +  │
+09)│  bigint_col@1 as sum_col] │
+10)└─────────────┬─────────────┘
+11)┌─────────────┴─────────────┐
+12)│      RepartitionExec      │
+13)└─────────────┬─────────────┘
+14)┌─────────────┴─────────────┐
+15)│       DataSourceExec      │
+16)│    --------------------   │
+17)│          files: 1         │
+18)│        format: csv        │
+19)└───────────────────────────┘
+
+
+# Query with projection on parquet
+query TT
+explain SELECT int_col, bigint_col, int_col+bigint_col AS sum_col FROM table2;
+----
+logical_plan
+01)Projection: table2.int_col, table2.bigint_col, CAST(table2.int_col AS Int64) + table2.bigint_col AS sum_col
+02)--TableScan: table2 projection=[int_col, bigint_col]
+physical_plan
+01)┌───────────────────────────┐
+02)│       ProjectionExec      │
+03)│    --------------------   │
+04)│   ProjectionExec: expr:   │
+05)│   [int_col@0 as int_col,  │
+06)│       bigint_col@1 as     │
+07)│      bigint_col, CAST     │
+08)│   (int_col@0 AS Int64) +  │
+09)│  bigint_col@1 as sum_col] │
+10)└─────────────┬─────────────┘
+11)┌─────────────┴─────────────┐
+12)│      RepartitionExec      │
+13)└─────────────┬─────────────┘
+14)┌─────────────┴─────────────┐
+15)│       DataSourceExec      │
+16)│    --------------------   │
+17)│          files: 1         │
+18)│      format: parquet      │
+19)└───────────────────────────┘
+
+
+# Query with projection on memory
+query TT
+explain SELECT int_col, bigint_col, int_col+bigint_col AS sum_col FROM table3;
+----
+logical_plan
+01)Projection: table3.int_col, table3.bigint_col, CAST(table3.int_col AS Int64) + table3.bigint_col AS sum_col
+02)--TableScan: table3 projection=[int_col, bigint_col]
+physical_plan
+01)┌───────────────────────────┐
+02)│       ProjectionExec      │
+03)│    --------------------   │
+04)│   ProjectionExec: expr:   │
+05)│   [int_col@0 as int_col,  │
+06)│       bigint_col@1 as     │
+07)│      bigint_col, CAST     │
+08)│   (int_col@0 AS Int64) +  │
+09)│  bigint_col@1 as sum_col] │
+10)└─────────────┬─────────────┘
+11)┌─────────────┴─────────────┐
+12)│       DataSourceExec      │
+13)│    --------------------   │
+14)│        bytes: 1560        │
+15)│       format: memory      │
+16)│          rows: 1          │
+17)└───────────────────────────┘
+
+
+
+# Query with projection on json
+query TT
+explain SELECT int_col, bigint_col, int_col+bigint_col AS sum_col FROM table4;
+----
+logical_plan
+01)Projection: table4.int_col, table4.bigint_col, table4.int_col + table4.bigint_col AS sum_col
+02)--TableScan: table4 projection=[bigint_col, int_col]
+physical_plan
+01)┌───────────────────────────┐
+02)│       ProjectionExec      │
+03)│    --------------------   │
+04)│   ProjectionExec: expr:   │
+05)│   [int_col@1 as int_col,  │
+06)│       bigint_col@0 as     │
+07)│     bigint_col, int_col   │
+08)│    @1 + bigint_col@0 as   │
+09)│          sum_col]         │
+10)└─────────────┬─────────────┘
+11)┌─────────────┴─────────────┐
+12)│      RepartitionExec      │
+13)└─────────────┬─────────────┘
+14)┌─────────────┴─────────────┐
+15)│       DataSourceExec      │
+16)│    --------------------   │
+17)│          files: 1         │
+18)│        format: json       │
+19)└───────────────────────────┘
+
+
+# Query with projection on arrow
+query TT
+explain SELECT int_col, bigint_col, int_col+bigint_col AS sum_col FROM table5;
+----
+logical_plan
+01)Projection: table5.int_col, table5.bigint_col, CAST(table5.int_col AS Int64) + table5.bigint_col AS sum_col
+02)--TableScan: table5 projection=[int_col, bigint_col]
+physical_plan
+01)┌───────────────────────────┐
+02)│       ProjectionExec      │
+03)│    --------------------   │
+04)│   ProjectionExec: expr:   │
+05)│   [int_col@0 as int_col,  │
+06)│       bigint_col@1 as     │
+07)│      bigint_col, CAST     │
+08)│   (int_col@0 AS Int64) +  │
+09)│  bigint_col@1 as sum_col] │
+10)└─────────────┬─────────────┘
+11)┌─────────────┴─────────────┐
+12)│      RepartitionExec      │
+13)└─────────────┬─────────────┘
+14)┌─────────────┴─────────────┐
+15)│       DataSourceExec      │
+16)│    --------------------   │
+17)│          files: 1         │
+18)│       format: arrow       │
+19)└───────────────────────────┘
+
 # cleanup
 statement ok
 drop table table1;