should be it

external/duckdb/test/sql/generated_columns/virtual/partition.test

@@ -0,0 +1,42 @@
+# name: test/sql/generated_columns/virtual/partition.test
+# description: Using a generated column as a PARTITION BY key
+# group: [virtual]
+
+statement ok
+PRAGMA enable_verification
+
+statement ok
+CREATE TABLE unit (
+	price INTEGER,
+	amount_sold INTEGER,
+	name VARCHAR,
+	total_profit AS (price * amount_sold)
+);
+
+statement ok
+INSERT INTO unit VALUES (5,4, 'Soda can')
+
+statement ok
+INSERT INTO unit VALUES (5,8, 'Mars bar')
+
+statement ok
+INSERT INTO unit VALUES (4, 5, 'Chewing gum')
+
+query IIII
+SELECT total_profit, COUNT(total_profit), SUM(amount_sold), SUM(price) FROM unit GROUP BY total_profit ORDER BY ALL
+----
+20	2	9	9
+40	1	8	5
+
+query IIIII
+SELECT total_profit,
+		name,
+		COUNT(total_profit) OVER(PARTITION BY total_profit) AS CountTotalProfit,
+		SUM(amount_sold) OVER(PARTITION BY total_profit) AS SumAmountSold,
+		SUM(price) OVER(PARTITION BY total_profit) AS SumPrice
+FROM unit
+ORDER BY 1, 2 DESC
+----
+20	Soda can	2	9	9
+20	Chewing gum	2	9	9
+40	Mars bar	1	8	5