Reformated unit test code to conform to PEP8.