For those curious, this adds the unittest as an example in the docs for `forward`.