how to test python code