Update write_test.py