Update all_tests.py