Update test_buffer.py