File: test_pivot_duplicate_aggregates.test

package info (click to toggle)
duckdb 1.5.1-3
  • links: PTS, VCS
  • area: main
  • in suites:
  • size: 299,196 kB
  • sloc: cpp: 865,414; ansic: 57,292; python: 18,871; sql: 12,663; lisp: 11,751; yacc: 7,412; lex: 1,682; sh: 747; makefile: 564
file content (100 lines) | stat: -rw-r--r-- 2,688 bytes parent folder | download | duplicates (4)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
# name: test/sql/pivot/test_pivot_duplicate_aggregates.test
# description: Test PIVOT with duplicate aggregate expressions
# group: [pivot]

# Test duplicate aggregates with 21+ values (original bug - issue #20187)
# This used to throw: "Pivot aggregate count mismatch (expected 2, found 1)"
statement ok
PIVOT (FROM range(21)) ON range USING sum(range), sum(range);

# Should have 42 columns (21 pivot values × 2 aggregates)
query I
SELECT COUNT(*) FROM (PIVOT (FROM range(21)) ON range USING sum(range), sum(range));
----
1

# Test with range(20)
statement ok
PIVOT (FROM range(20)) ON range USING sum(range), sum(range);

# Test with smaller range
statement ok
PIVOT (FROM range(10)) ON range USING sum(range), sum(range);

# Test with three duplicate aggregates
statement ok
PIVOT (FROM range(10)) ON range USING sum(range), sum(range), sum(range);

# Verify three duplicates create 3x columns (30 columns = 10 values × 3 aggregates)
query I
SELECT COUNT(*) FROM (PIVOT (FROM range(10)) ON range USING sum(range), sum(range), sum(range));
----
1

# Test mix of duplicate and unique aggregates
statement ok
PIVOT (FROM range(21)) ON range USING sum(range), max(range), sum(range);

# should have 63 columns (21 values × 3 aggregates)
query I
SELECT COUNT(*) FROM (PIVOT (FROM range(21)) ON range USING sum(range), max(range), sum(range));
----
1

# Test with complex expressions
statement ok
PIVOT (FROM range(20)) ON range USING sum(range), sum(range);

# Test different aggregate functions
statement ok
PIVOT (FROM range(21)) ON range USING avg(range), avg(range);

statement ok
PIVOT (FROM range(21)) ON range USING count(range), count(range);

# Test with actual data to verify correctness
query IIIIII
SELECT * FROM (
  PIVOT (FROM range(3)) ON range USING sum(range), sum(range)
);
----
0	0	1	1	2	2

# Test that values are correct
query II
SELECT "0_sum(""range"")", "0_sum(""range"")_1" FROM (
  PIVOT (FROM range(21)) ON range USING sum(range), sum(range)
);
----
0	0

query II
SELECT "5_sum(""range"")", "5_sum(""range"")_1" FROM (
  PIVOT (FROM range(21)) ON range USING sum(range), sum(range)
);
----
5	5

# Test with mixed aggregates to ensure correct mapping
query III
SELECT "0_sum(""range"")", "0_max(""range"")", "0_sum(""range"")_1" FROM (
  PIVOT (FROM range(21)) ON range USING sum(range), max(range), sum(range)
);
----
0	0	0

query III
SELECT "5_sum(""range"")", "5_max(""range"")", "5_sum(""range"")_1" FROM (
  PIVOT (FROM range(21)) ON range USING sum(range), max(range), sum(range)
);
----
5	5	5

# Test with string aggregates
statement ok
PIVOT (
  SELECT range, range::VARCHAR as str_range
  FROM range(21)
)
ON range
USING max(str_range), max(str_range);