Removed unused import of Mock.
Added PEP8 testing.