Move the example to test.py