Adds more docs to the pytest example code

Adds more docs to the pytest example code #1264