Update test-driven.py