Add missing mock in test reqs
Fix pep8 issue