Add more testing cases

Author: zhuqi-lucas

datafusion/sqllogictest/test_files/aggregate_skip_partial.slt

@@ -220,6 +220,9 @@ set datafusion.execution.batch_size = 4;
 
 # Inserting into nullable table with batch_size specified above
 # to prevent creation on single in-memory batch
+
+# Test error case: the error case is expected here, as the table c5 is not nullable
+# and the query tries to insert a nullable value
 statement ok
 CREATE TABLE aggregate_test_100_null (
   c2  TINYINT NOT NULL,
@@ -237,6 +240,27 @@ SELECT
   CASE WHEN c1 = 'a' THEN NULL ELSE c11 END as c11
 FROM aggregate_test_100;
 
+statement ok
+drop table aggregate_test_100_null;
+
+# Test successful insert case which c5 is nullable
+statement ok
+CREATE TABLE aggregate_test_100_null (
+  c2  TINYINT NOT NULL,
+  c5  INT,
+  c3  SMALLINT,
+  c11 FLOAT
+);
+
+statement ok
+INSERT INTO aggregate_test_100_null
+SELECT
+  c2,
+  c5,
+  CASE WHEN c1 = 'e' THEN NULL ELSE c3 END as c3,
+  CASE WHEN c1 = 'a' THEN NULL ELSE c11 END as c11
+FROM aggregate_test_100;
+
 # Test count varchar / int / float
 query IIII
 SELECT c2, count(c1), count(c5), count(c11) FROM aggregate_test_100 GROUP BY c2 ORDER BY c2;
@@ -301,11 +325,21 @@ SELECT c2, approx_distinct(c1), approx_distinct(c5) FROM aggregate_test_100 GROU
 query III
 SELECT c2, count(c3), count(c11) FROM aggregate_test_100_null GROUP BY c2 ORDER BY c2;
 ----
+1 19 17
+2 17 19
+3 15 13
+4 16 19
+5 12 11
 
 # Test min / max with nullable fields
 query IIIRR
 SELECT c2, min(c3), max(c3), min(c11), max(c11) FROM aggregate_test_100_null GROUP BY c2 ORDER BY c2;
 ----
+1 -99 125 0.064453244 0.89651865
+2 -117 122 0.09683716 0.8315913
+3 -101 123 0.034291923 0.94669616
+4 -117 123 0.028003037 0.7085086
+5 -101 118 0.12559289 0.87989986
 
 # Test sum with nullable fields
 query IIR
@@ -321,16 +355,31 @@ SELECT c2, sum(c3), sum(c11) FROM aggregate_test_100 GROUP BY c2 ORDER BY c2;
 query IIR
 SELECT c2, median(c3), median(c11) FROM aggregate_test_100_null GROUP BY c2 ORDER BY c2;
 ----
+1 12 0.6067944
+2 1 0.46076488
+3 14 0.40154034
+4 -17 0.48515016
+5 -35 0.5536642
 
 # Test approx_median with nullable fields
 query IIR
 SELECT c2, approx_median(c3), approx_median(c11) FROM aggregate_test_100_null GROUP BY c2 ORDER BY c2;
 ----
+1 12 0.6067944
+2 1 0.46076488
+3 14 0.40154034
+4 -7 0.48515016
+5 -39 0.5536642
 
 # Test approx_distinct with nullable fields
 query II
 SELECT c2, approx_distinct(c3) FROM aggregate_test_100_null GROUP BY c2 ORDER BY c2;
 ----
+1 19
+2 16
+3 13
+4 16
+5 12
 
 # Test avg for tinyint / float
 query TRR
@@ -467,6 +516,11 @@ SELECT c2,
        COUNT(c11) FILTER(WHERE c5 > 0)
 FROM aggregate_test_100_null GROUP BY c2 ORDER BY c2;
 ----
+1 11 6
+2 6 6
+3 8 6
+4 11 14
+5 8 7
 
 # Test avg for tinyint / float
 query TRR
@@ -489,6 +543,11 @@ SELECT c2,
        COUNT(c11) FILTER(WHERE c3 > 0)
 FROM aggregate_test_100_null GROUP BY c2 ORDER BY c2;
 ----
+1 10 9
+2 7 8
+3 3 6
+4 3 7
+5 6 3
 
 # Test min / max with nullable fields and filter
 query IIIRR
@@ -499,6 +558,11 @@ SELECT c2,
        MAX(c11) FILTER (WHERE c5 < 0)
 FROM aggregate_test_100_null GROUP BY c2 ORDER BY c2;
 ----
+1 -99 103 0.2578469 0.89651865
+2 -48 93 0.09683716 0.8315913
+3 -76 123 0.034291923 0.94669616
+4 -117 123 0.06563997 0.57360977
+5 -94 68 0.12559289 0.75173044
 
 # Test min / max with nullable fields and nullable filter
 query III
@@ -507,6 +571,11 @@ SELECT c2,
        MAX(c3) FILTER (WHERE c11 > 0.5)
 FROM aggregate_test_100_null GROUP BY c2 ORDER BY c2;
 ----
+1 -99 125
+2 -106 122
+3 -76 73
+4 -117 47
+5 -82 118
 
 # Test sum with nullable field and nullable / non-nullable filters
 query IIIRR
@@ -517,20 +586,35 @@ SELECT c2,
        SUM(c11) FILTER (WHERE c3 > 0)
 FROM aggregate_test_100_null GROUP BY c2 ORDER BY c2;
 ----
+1 -3 77 7.214695632458 5.085060358047
+2 100 77 6.197732746601 3.150197088718
+3 109 211 2.80575042963 2.80632930994
+4 -171 56 2.10740506649 1.939846396446
+5 -86 -76 1.8741710186 1.600569307804
 
 # Test approx_distinct with nullable fields and filter
 query II
 SELECT c2,
        approx_distinct(c3) FILTER (WHERE c5 > 0)
 FROM aggregate_test_100_null GROUP BY c2 ORDER BY c2;
 ----
+1 11
+2 6
+3 6
+4 11
+5 8
 
 # Test approx_distinct with nullable fields and nullable filter
 query II
 SELECT c2,
        approx_distinct(c3) FILTER (WHERE c11 > 0.5)
 FROM aggregate_test_100_null GROUP BY c2 ORDER BY c2;
 ----
+1 10
+2 6
+3 3
+4 3
+5 6
 
 # Test median with nullable fields and filter
 query IIR
@@ -539,13 +623,23 @@ SELECT c2,
        median(c11) FILTER (WHERE c5 < 0)
 FROM aggregate_test_100_null GROUP BY c2 ORDER BY c2;
 ----
+1 -5 0.6623719
+2 15 0.52930677
+3 13 0.32792538
+4 -38 0.49774808
+5 -18 0.49842384
 
 # Test min / max with nullable fields and nullable filter
 query II
 SELECT c2,
        median(c3) FILTER (WHERE c11 > 0.5)
 FROM aggregate_test_100_null GROUP BY c2 ORDER BY c2;
 ----
+1 33
+2 -29
+3 22
+4 -90
+5 -22
 
 # Test approx_median with nullable fields and filter
 query IIR
@@ -554,13 +648,23 @@ SELECT c2,
        approx_median(c11) FILTER (WHERE c5 < 0)
 FROM aggregate_test_100_null GROUP BY c2 ORDER BY c2;
 ----
+1 -5 0.6623719
+2 12 0.52930677
+3 13 0.32792538
+4 -38 0.49774808
+5 -21 0.47652745
 
 # Test approx_median with nullable fields and nullable filter
 query II
 SELECT c2,
        approx_median(c3) FILTER (WHERE c11 > 0.5)
 FROM aggregate_test_100_null GROUP BY c2 ORDER BY c2;
 ----
+1 35
+2 -29
+3 22
+4 -90
+5 -32
 
 statement ok
 DROP TABLE aggregate_test_100_null;