Fixed import path in test case.
Some PEP8 formatting