| import tempfile | |
| import pytest | |
| import os | |
| from os import path | |
| from ding.data.storage import FileStorage | |
| 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_) | |