add test.py (for me only)