Mockito: как имитировать doAnswer для универсального метода

У меня есть класс, который, по сути, выглядит так:

class Checkpointer {
  public <Input,Output> Output runFunction(Input input, Function<Input,Output> function) {
    Output output;
    // Sometimes run the function, sometimes return an Output from a cache
    return output
  }
}

Я хотел бы издеваться над этим классом, используя Mockito doAnswer:

Checkpointer checkpointer; // mocked via @Mock annotation

Mockito
  .doAnswer(/* ??? */)
  .when(checkpointer)
  .runFunction(Mockito.any(), Mockito.any());

Функция, которую я хочу имитировать, должна быть универсальной. Можно ли это сделать?

Например, моя первая попытка выдала следующее. Мало того, что я прибегал к Object в качестве аргументов типа для Function, так еще и компилятор был недоволен непроверенным приведением типов:

Mockito.doAnswer((invocation) ->
        {
            // compiler is not happy with this cast   V
            Function<Object,Object> function = (Function<Object,Object>)invocation.getArguments()[1]; 
            return function.apply(invocation.getArgument(0));
        }).when(checkpointer).runFunction(Mockito.any(), Mockito.any());

Если это невозможно сделать, я думаю, можно попробовать написать свой собственный фиктивный класс, расширяющий первый, и использовать Mockito.spy.