Add a test python