Got Python testing framework working