Update WriteDataTest.py