Corrected testcase and pep8

Change-Id: Ie68ae4aa71e0541d2052d1727490481e3a51be6d