Ensure test example follows PEP 8 guidelines

Given the nature of this project we should follow PEP 8 guidelines where possible.

PEP 8 specifies that standard library and local imports are separated by a blank line. Furthermore, top-level declarations are separated by two blank lines and method definitions by one blank line (I presume this also applies to the class definition and method definitions).