add some array test python code