Skip to content

Commit a7d919f

Browse files
committed
Add integration tests for correlation, covariance, regression, and moments functions
1 parent 085ac9d commit a7d919f

1 file changed

Lines changed: 42 additions & 42 deletions

File tree

  • iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/execution/operator/source/relational/aggregation

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/execution/operator/source/relational/aggregation/AccumulatorFactory.java

Lines changed: 42 additions & 42 deletions
Original file line numberDiff line numberDiff line change
@@ -471,10 +471,10 @@ public static TableAccumulator createBuiltinMultiInputAccumulator(
471471
switch (aggregationType) {
472472
case MAX_BY:
473473
checkState(inputDataTypes.size() == 2, "Wrong inputDataTypes size.");
474-
// return new MaxByAccumulator(inputDataTypes.get(0), inputDataTypes.get(1));
474+
// return new MaxByAccumulator(inputDataTypes.get(0), inputDataTypes.get(1));
475475
case MIN_BY:
476476
checkState(inputDataTypes.size() == 2, "Wrong inputDataTypes size.");
477-
// return new MinByAccumulator(inputDataTypes.get(0), inputDataTypes.get(1));
477+
// return new MinByAccumulator(inputDataTypes.get(0), inputDataTypes.get(1));
478478
default:
479479
throw new IllegalArgumentException("Invalid Aggregation function: " + aggregationType);
480480
}
@@ -491,46 +491,46 @@ private static TableAccumulator createBuiltinSingleInputAccumulator(
491491
return new CountAccumulator();
492492
case AVG:
493493
return new AvgAccumulator(tsDataType);
494-
/*case SUM:
495-
return new SumAccumulator(tsDataType);
496-
case EXTREME:
497-
return new ExtremeAccumulator(tsDataType);
498-
case MAX_TIME:
499-
return ascending ? new MaxTimeAccumulator() : new MaxTimeDescAccumulator();
500-
case MIN_TIME:
501-
return ascending ? new MinTimeAccumulator() : new MinTimeDescAccumulator();
502-
case MAX_VALUE:
503-
return new MaxValueAccumulator(tsDataType);
504-
case MIN_VALUE:
505-
return new MinValueAccumulator(tsDataType);
506-
case LAST_VALUE:
507-
return ascending
508-
? new LastValueAccumulator(tsDataType)
509-
: new LastValueDescAccumulator(tsDataType);
510-
case FIRST_VALUE:
511-
return ascending
512-
? new FirstValueAccumulator(tsDataType)
513-
: new FirstValueDescAccumulator(tsDataType);
514-
case COUNT_IF:
515-
return new CountIfAccumulator(
516-
initKeepEvaluator(inputExpressions.get(1)),
517-
Boolean.parseBoolean(inputAttributes.getOrDefault("ignoreNull", "true")));
518-
case TIME_DURATION:
519-
return new TimeDurationAccumulator();
520-
case MODE:
521-
return createModeAccumulator(tsDataType);
522-
case COUNT_TIME:
523-
return new CountTimeAccumulator();
524-
case STDDEV:
525-
case STDDEV_SAMP:
526-
return new VarianceAccumulator(tsDataType, VarianceAccumulator.VarianceType.STDDEV_SAMP);
527-
case STDDEV_POP:
528-
return new VarianceAccumulator(tsDataType, VarianceAccumulator.VarianceType.STDDEV_POP);
529-
case VARIANCE:
530-
case VAR_SAMP:
531-
return new VarianceAccumulator(tsDataType, VarianceAccumulator.VarianceType.VAR_SAMP);
532-
case VAR_POP:
533-
return new VarianceAccumulator(tsDataType, VarianceAccumulator.VarianceType.VAR_POP);*/
494+
/*case SUM:
495+
return new SumAccumulator(tsDataType);
496+
case EXTREME:
497+
return new ExtremeAccumulator(tsDataType);
498+
case MAX_TIME:
499+
return ascending ? new MaxTimeAccumulator() : new MaxTimeDescAccumulator();
500+
case MIN_TIME:
501+
return ascending ? new MinTimeAccumulator() : new MinTimeDescAccumulator();
502+
case MAX_VALUE:
503+
return new MaxValueAccumulator(tsDataType);
504+
case MIN_VALUE:
505+
return new MinValueAccumulator(tsDataType);
506+
case LAST_VALUE:
507+
return ascending
508+
? new LastValueAccumulator(tsDataType)
509+
: new LastValueDescAccumulator(tsDataType);
510+
case FIRST_VALUE:
511+
return ascending
512+
? new FirstValueAccumulator(tsDataType)
513+
: new FirstValueDescAccumulator(tsDataType);
514+
case COUNT_IF:
515+
return new CountIfAccumulator(
516+
initKeepEvaluator(inputExpressions.get(1)),
517+
Boolean.parseBoolean(inputAttributes.getOrDefault("ignoreNull", "true")));
518+
case TIME_DURATION:
519+
return new TimeDurationAccumulator();
520+
case MODE:
521+
return createModeAccumulator(tsDataType);
522+
case COUNT_TIME:
523+
return new CountTimeAccumulator();
524+
case STDDEV:
525+
case STDDEV_SAMP:
526+
return new VarianceAccumulator(tsDataType, VarianceAccumulator.VarianceType.STDDEV_SAMP);
527+
case STDDEV_POP:
528+
return new VarianceAccumulator(tsDataType, VarianceAccumulator.VarianceType.STDDEV_POP);
529+
case VARIANCE:
530+
case VAR_SAMP:
531+
return new VarianceAccumulator(tsDataType, VarianceAccumulator.VarianceType.VAR_SAMP);
532+
case VAR_POP:
533+
return new VarianceAccumulator(tsDataType, VarianceAccumulator.VarianceType.VAR_POP);*/
534534
default:
535535
throw new IllegalArgumentException("Invalid Aggregation function: " + aggregationType);
536536
}

0 commit comments

Comments
 (0)