Use new formatting in test.py