Cleanup temporary files in unit tests
PEP8