Switch test commands to python