made test.py comply with pep8