如何测试将json写入磁盘的python函数？

import json
import os

def write_json_to_disk(data, file_path):
    with open(file_path, 'w') as file:
        json.dump(data, file)

def test_write_json_to_disk():
    data = {'name': 'John', 'age': 30, 'city': 'New York'}
    file_path = 'data.json'
    
    # 调用函数将JSON写入磁盘
    write_json_to_disk(data, file_path)
    
    # 验证文件是否存在
    assert os.path.exists(file_path)
    
    # 验证文件内容是否正确
    with open(file_path, 'r') as file:
        saved_data = json.load(file)
        assert saved_data == data
    
    # 清理测试生成的文件
    os.remove(file_path)

test_write_json_to_disk()