improve test.py