python code for test