def test_resource_fingerprinting():
resource1 = Resource("resource 1")
resource2 = Resource("resource 2")
fingerprint1 = resource1.fingerprint()
fingerprint2 = resource2.fingerprint()
assert fingerprint1
assert fingerprint2
assert fingerprint1 != fingerprint2
def test_resource_fingerprinting():
resource1a = Resource("resource 1")
resource1b = Resource("resource 1")
resource2 = Resource("resource 3")
fingerprint1 = resource1a.fingerprint()
fingerprint2 = resource1b.fingerprint()
fingerprint3 = resource2.fingerprint()
assert fingerprint1
assert fingerprint2
assert fingerprint3
assert fingerprint1 != fingerprint2
assert fingerprint2 != fingerprint3
assert fingerprint1 != fingerprint3
def test_resource_caching(tmp_path_factory: TempPathFactory):
model_storage = LocalModelStorage(
tmp_path_factory.mktemp("initial_model_storage"))
resource = Resource("my resource")
# Fill model storage
test_filename = "file.txt"
test_content = "test_resource_caching"
with model_storage.write_to(resource) as temporary_directory:
file = temporary_directory / test_filename
file.write_text(test_content)
cache_dir = tmp_path_factory.mktemp("cache_dir")
# Cache resource
resource.to_cache(cache_dir, model_storage)
# Reload resource from cache and inspect
new_model_storage = LocalModelStorage(
tmp_path_factory.mktemp("new_model_storage"))
reinstantiated_resource = Resource.from_cache(resource.name, cache_dir,
new_model_storage,
resource.output_fingerprint)
assert reinstantiated_resource == resource
assert reinstantiated_resource.fingerprint() == resource.fingerprint()
# Read written resource data from model storage to see whether all expected
# contents are there
with new_model_storage.read_from(resource) as temporary_directory:
assert (temporary_directory /
test_filename).read_text() == test_content
