Indent code in test.py properly