clean up, exception assertion in test.py