from pydantic_factories import ModelFactory from pynyaata.cache import client from pynyaata.types import RemoteFile class RemoteFileFactory(ModelFactory[RemoteFile]): __model__ = RemoteFile def test_cache_data(): remote = RemoteFileFactory.build() client.set("test", remote) assert client.get("test") == remote