Update setup.py and test runner