Ensure test example follows PEP 8 guidelines

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).