Add test.py (same file as on Python API sites page)