Update test_format.py