TST: Fix tests for test_format.py