DI-engine/ding/data/storage/tests/test_storage.py

import tempfile
import pytest
import os
from os import path
from ding.data.storage import FileStorage


@pytest.mark.unittest
def test_file_storage():
    path_ = path.join(tempfile.gettempdir(), "test_storage.txt")
    try:
        storage = FileStorage(path=path_)
        storage.save("test")
        content = storage.load()
        assert content == "test"
    finally:
        if path.exists(path_):
            os.remove(path_)