clean test.py