package io.burt.jmespath.function;

import java.util.List;
import java.util.ArrayList;

import io.burt.jmespath.Adapter;

public class AvgFunction extends JmesPathFunction {
  public AvgFunction() {
    super(1, 1);
  }

  @Override
  protected <T> T internalCall(Adapter<T> adapter, List<ExpressionOrValue<T>> arguments) {
    T array = arguments.get(0).value();
    if (isNumberArray(adapter, array)) {
      double sum = 0;
      int count = 0;
      for (T n : adapter.toList(array)) {
        sum += adapter.toDouble(n);
        count += 1;
      }
      return adapter.createNumber(sum/count);
    } else {
      List<T> values = adapter.toList(array);
      List<String> types = new ArrayList<>(values.size());
      for (T value : values) {
        types.add(adapter.typeOf(value));
      }
      throw new ArgumentTypeException(name(), "array of numbers", types.toString());
    }
  }

  private <T> boolean isNumberArray(Adapter<T> adapter, T array) {
    if (adapter.typeOf(array).equals("array")) {
      for (T element : adapter.toList(array)) {
        if (!adapter.isNumber(element)) {
          return false;
        }
      }
      return true;
    } else {
      return false;
    }
  }
}

@@ -975,36 +975,21 @@
 ter.
-typeOf(array).equals(%22
+isArray(
 array
-%22
 )) %7B

Use Adapter#isArray and not #typeOf in AvgFunction