Adding Python test code for local use only. will put it on RG soon