Migrate aggregation function tests to JUnit

core/trino-main/src/test/java/io/trino/operator/TestRealAverageAggregation.java

@@ -19,7 +19,7 @@
 import io.trino.spi.block.BlockBuilder;
 import io.trino.spi.type.Type;
 import io.trino.sql.tree.QualifiedName;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.List;
 
@@ -29,7 +29,6 @@
 import static io.trino.sql.analyzer.TypeSignatureProvider.fromTypes;
 import static java.lang.Float.floatToRawIntBits;
 
-@Test(singleThreaded = true)
 public class TestRealAverageAggregation
         extends AbstractTestAggregationFunction
 {

core/trino-main/src/test/java/io/trino/operator/aggregation/AbstractTestAggregationFunction.java

@@ -28,7 +28,7 @@
 import io.trino.spi.function.WindowIndex;
 import io.trino.spi.type.Type;
 import io.trino.sql.tree.QualifiedName;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.lang.reflect.Constructor;
 import java.util.List;

core/trino-main/src/test/java/io/trino/operator/aggregation/AbstractTestApproximateSetGeneric.java

@@ -25,7 +25,7 @@
 import io.trino.spi.type.Type;
 import io.trino.sql.tree.QualifiedName;
 import org.apache.commons.math3.stat.descriptive.DescriptiveStatistics;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.ArrayList;
 import java.util.Collections;

core/trino-main/src/test/java/io/trino/operator/aggregation/TestAccumulatorCompiler.java

@@ -35,7 +35,7 @@
 import io.trino.spi.type.RealType;
 import io.trino.spi.type.TimestampType;
 import io.trino.sql.gen.IsolatedClass;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.lang.invoke.MethodHandle;
 import java.lang.reflect.Constructor;

core/trino-main/src/test/java/io/trino/operator/aggregation/TestAnyValueAggregation.java

@@ -21,7 +21,7 @@
 import io.trino.spi.type.Type;
 import io.trino.spi.type.TypeOperators;
 import io.trino.sql.tree.QualifiedName;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.Arrays;
 import java.util.Collection;

core/trino-main/src/test/java/io/trino/operator/aggregation/TestApproximateCountDistinctAggregations.java

@@ -13,7 +13,7 @@
  */
 package io.trino.operator.aggregation;
 
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import static io.trino.operator.aggregation.ApproximateCountDistinctAggregation.standardErrorToBuckets;
 import static io.trino.spi.StandardErrorCode.INVALID_FUNCTION_ARGUMENT;

core/trino-main/src/test/java/io/trino/operator/aggregation/TestApproximateMostFrequentHistogram.java

@@ -16,7 +16,7 @@
 import com.google.common.collect.ImmutableMap;
 import io.airlift.slice.Slice;
 import io.airlift.slice.Slices;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.Map;
 

core/trino-main/src/test/java/io/trino/operator/aggregation/TestApproximatePercentileAggregation.java

@@ -22,7 +22,7 @@
 import io.trino.spi.type.ArrayType;
 import io.trino.sql.analyzer.TypeSignatureProvider;
 import io.trino.sql.tree.QualifiedName;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.List;
 

core/trino-main/src/test/java/io/trino/operator/aggregation/TestArbitraryAggregation.java

@@ -21,7 +21,7 @@
 import io.trino.spi.type.Type;
 import io.trino.spi.type.TypeOperators;
 import io.trino.sql.tree.QualifiedName;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.Arrays;
 import java.util.Collection;

core/trino-main/src/test/java/io/trino/operator/aggregation/TestArrayAggregation.java

@@ -23,7 +23,7 @@
 import io.trino.spi.type.ArrayType;
 import io.trino.spi.type.SqlDate;
 import io.trino.sql.tree.QualifiedName;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.ArrayList;
 import java.util.Arrays;

core/trino-main/src/test/java/io/trino/operator/aggregation/TestBitwiseAndAggregation.java

@@ -17,7 +17,7 @@
 import io.trino.spi.block.Block;
 import io.trino.spi.block.BlockBuilder;
 import io.trino.spi.type.Type;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.List;
 import java.util.stream.LongStream;

core/trino-main/src/test/java/io/trino/operator/aggregation/TestBitwiseOrAggregation.java

@@ -17,7 +17,7 @@
 import io.trino.spi.block.Block;
 import io.trino.spi.block.BlockBuilder;
 import io.trino.spi.type.Type;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.List;
 import java.util.stream.LongStream;

core/trino-main/src/test/java/io/trino/operator/aggregation/TestChecksumAggregation.java

@@ -22,7 +22,7 @@
 import io.trino.sql.tree.QualifiedName;
 import io.trino.type.BlockTypeOperators;
 import io.trino.type.BlockTypeOperators.BlockPositionXxHash64;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import static io.airlift.slice.Slices.wrappedLongArray;
 import static io.trino.block.BlockAssertions.createArrayBigintBlock;

core/trino-main/src/test/java/io/trino/operator/aggregation/TestDoubleCorrelationAggregation.java

@@ -17,7 +17,7 @@
 import io.trino.spi.block.Block;
 import io.trino.spi.type.Type;
 import org.apache.commons.math3.stat.correlation.PearsonsCorrelation;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.List;
 import java.util.stream.DoubleStream;

core/trino-main/src/test/java/io/trino/operator/aggregation/TestDoubleHistogramAggregation.java

@@ -24,7 +24,7 @@
 import io.trino.spi.type.Type;
 import io.trino.sql.planner.plan.AggregationNode.Step;
 import io.trino.sql.tree.QualifiedName;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.Map;
 import java.util.OptionalInt;

core/trino-main/src/test/java/io/trino/operator/aggregation/TestDoubleRegrInterceptAggregation.java

@@ -17,7 +17,7 @@
 import io.trino.spi.block.Block;
 import io.trino.spi.type.Type;
 import org.apache.commons.math3.stat.regression.SimpleRegression;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.List;
 

core/trino-main/src/test/java/io/trino/operator/aggregation/TestDoubleRegrSlopeAggregation.java

@@ -17,7 +17,7 @@
 import io.trino.spi.block.Block;
 import io.trino.spi.type.Type;
 import org.apache.commons.math3.stat.regression.SimpleRegression;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.List;
 

core/trino-main/src/test/java/io/trino/operator/aggregation/TestHistogram.java

@@ -31,7 +31,7 @@
 import io.trino.sql.tree.QualifiedName;
 import org.joda.time.DateTime;
 import org.joda.time.DateTimeZone;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.ArrayList;
 import java.util.HashMap;

core/trino-main/src/test/java/io/trino/operator/aggregation/TestListagg.java

@@ -15,7 +15,7 @@
 
 import io.trino.metadata.TestingFunctionResolution;
 import io.trino.sql.tree.QualifiedName;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.Collections;
 

core/trino-main/src/test/java/io/trino/operator/aggregation/TestMapAggAggregation.java

@@ -21,7 +21,7 @@
 import io.trino.spi.type.MapType;
 import io.trino.spi.type.RowType;
 import io.trino.sql.tree.QualifiedName;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.LinkedHashMap;
 import java.util.Map;

core/trino-main/src/test/java/io/trino/operator/aggregation/TestMapUnionAggregation.java

@@ -19,7 +19,7 @@
 import io.trino.spi.type.ArrayType;
 import io.trino.spi.type.MapType;
 import io.trino.sql.tree.QualifiedName;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.HashMap;
 import java.util.Map;

core/trino-main/src/test/java/io/trino/operator/aggregation/TestMergeQuantileDigestFunction.java

@@ -24,7 +24,7 @@
 import io.trino.spi.type.TypeSignature;
 import io.trino.spi.type.TypeSignatureParameter;
 import io.trino.sql.tree.QualifiedName;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.List;
 import java.util.function.BiFunction;

core/trino-main/src/test/java/io/trino/operator/aggregation/TestMinMaxByAggregation.java

@@ -23,7 +23,7 @@
 import io.trino.spi.type.TypeOperators;
 import io.trino.sql.analyzer.TypeSignatureProvider;
 import io.trino.sql.tree.QualifiedName;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.Collection;
 import java.util.List;

core/trino-main/src/test/java/io/trino/operator/aggregation/TestMultimapAggAggregation.java

@@ -29,7 +29,7 @@
 import io.trino.spi.type.RowType;
 import io.trino.spi.type.Type;
 import io.trino.sql.tree.QualifiedName;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.ArrayList;
 import java.util.HashMap;

core/trino-main/src/test/java/io/trino/operator/aggregation/TestNumericHistogram.java

@@ -16,7 +16,7 @@
 import com.google.common.collect.ImmutableMap;
 import com.google.common.collect.Maps;
 import io.airlift.slice.Slice;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.HashSet;
 import java.util.Map;

core/trino-main/src/test/java/io/trino/operator/aggregation/TestRealHistogramAggregation.java

@@ -23,7 +23,7 @@
 import io.trino.spi.type.MapType;
 import io.trino.sql.planner.plan.AggregationNode.Step;
 import io.trino.sql.tree.QualifiedName;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.Map;
 import java.util.OptionalInt;

core/trino-main/src/test/java/io/trino/operator/aggregation/TestRealRegrInterceptAggregation.java

@@ -17,7 +17,7 @@
 import io.trino.spi.block.Block;
 import io.trino.spi.type.Type;
 import org.apache.commons.math3.stat.regression.SimpleRegression;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.List;
 

core/trino-main/src/test/java/io/trino/operator/aggregation/TestRealRegrSlopeAggregation.java

@@ -17,7 +17,7 @@
 import io.trino.spi.block.Block;
 import io.trino.spi.type.Type;
 import org.apache.commons.math3.stat.regression.SimpleRegression;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.List;
 

core/trino-main/src/test/java/io/trino/operator/aggregation/TestTypedHistogram.java

@@ -21,7 +21,7 @@
 import io.trino.spi.type.MapType;
 import io.trino.spi.type.TypeOperators;
 import io.trino.type.BlockTypeOperators;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.function.IntUnaryOperator;
 import java.util.stream.IntStream;

core/trino-main/src/test/java/io/trino/operator/aggregation/TestTypedSet.java

@@ -20,7 +20,7 @@
 import io.trino.spi.type.Type;
 import io.trino.spi.type.TypeOperators;
 import io.trino.type.BlockTypeOperators;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.HashSet;
 import java.util.List;

core/trino-main/src/test/java/io/trino/operator/aggregation/listagg/TestListaggAggregationFunction.java

@@ -22,7 +22,7 @@
 import io.trino.spi.block.VariableWidthBlockBuilder;
 import io.trino.sql.analyzer.TypeSignatureProvider;
 import io.trino.sql.tree.QualifiedName;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.nio.charset.StandardCharsets;
 import java.util.List;

core/trino-main/src/test/java/io/trino/operator/aggregation/minmaxbyn/TestMinMaxByNAggregation.java

@@ -21,7 +21,7 @@
 import io.trino.spi.type.ArrayType;
 import io.trino.sql.analyzer.TypeSignatureProvider;
 import io.trino.sql.tree.QualifiedName;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.Arrays;
 import java.util.List;

core/trino-main/src/test/java/io/trino/operator/aggregation/minmaxbyn/TestTypedKeyValueHeap.java

@@ -16,7 +16,7 @@
 import io.trino.spi.block.Block;
 import io.trino.spi.block.BlockBuilder;
 import io.trino.spi.type.TypeOperators;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.lang.invoke.MethodHandle;
 import java.util.ArrayList;

core/trino-main/src/test/java/io/trino/operator/aggregation/minmaxn/TestArrayMaxNAggregation.java

@@ -20,7 +20,7 @@
 import io.trino.spi.block.Block;
 import io.trino.spi.type.ArrayType;
 import io.trino.spi.type.Type;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.Arrays;
 import java.util.List;

core/trino-main/src/test/java/io/trino/operator/aggregation/minmaxn/TestDoubleMinNAggregation.java

@@ -17,7 +17,7 @@
 import io.trino.operator.aggregation.AbstractTestAggregationFunction;
 import io.trino.spi.block.Block;
 import io.trino.spi.type.Type;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.Arrays;
 import java.util.List;

core/trino-main/src/test/java/io/trino/operator/aggregation/minmaxn/TestLongMaxNAggregation.java

@@ -17,7 +17,7 @@
 import io.trino.operator.aggregation.AbstractTestAggregationFunction;
 import io.trino.spi.block.Block;
 import io.trino.spi.type.Type;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.util.Arrays;
 import java.util.List;

core/trino-main/src/test/java/io/trino/operator/aggregation/minmaxn/TestTypedHeap.java

@@ -17,7 +17,7 @@
 import io.trino.spi.block.BlockBuilder;
 import io.trino.spi.type.Type;
 import io.trino.spi.type.TypeOperators;
-import org.testng.annotations.Test;
+import org.junit.jupiter.api.Test;
 
 import java.lang.invoke.MethodHandle;
 import java.util.ArrayList;