Add diff to test.py