File size: 435 Bytes
079c32c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
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_)
|