PEP8 changes for test_basic.py