Skip to content

Commit 2bce613

Browse files
committed
Add end-to-end tests for numeric_histogram aggregate function
1 parent ac765b9 commit 2bce613

File tree

1 file changed

+17
-0
lines changed

1 file changed

+17
-0
lines changed

presto-native-execution/src/test/java/com/facebook/presto/nativeworker/AbstractTestNativeAggregations.java

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -354,6 +354,23 @@ public void testMultiMapAgg()
354354
assertQuery("SELECT orderkey, multimap_agg(linenumber % 3, discount) FROM lineitem GROUP BY 1");
355355
}
356356

357+
@Test
358+
public void testNumericHistogram()
359+
{
360+
// numeric_histogram with 2 arguments (buckets, value)
361+
assertQuerySucceeds("SELECT numeric_histogram(5, quantity) FROM lineitem");
362+
assertQuerySucceeds("SELECT orderkey, numeric_histogram(3, quantity) FROM lineitem GROUP BY 1 LIMIT 10");
363+
364+
// numeric_histogram with 3 arguments (buckets, value, weight)
365+
assertQuerySucceeds("SELECT numeric_histogram(5, quantity, 1.0) FROM lineitem");
366+
assertQuerySucceeds("SELECT numeric_histogram(10, extendedprice, quantity) FROM lineitem");
367+
assertQuerySucceeds("SELECT orderkey, numeric_histogram(5, quantity, 1.0) FROM lineitem GROUP BY 1 LIMIT 10");
368+
369+
// Test with real type
370+
assertQuerySucceeds("SELECT numeric_histogram(5, tax_as_real) FROM lineitem");
371+
assertQuerySucceeds("SELECT numeric_histogram(5, tax_as_real, 1.0) FROM lineitem");
372+
}
373+
357374
@Test(dataProvider = "exchangeEncodingProvider")
358375
public void testMarkDistinct(String exchangeEncoding)
359376
{

0 commit comments

Comments
 (0)