cleanup test.py