wehub-resource-sync
29 lines
842 B
Python
29 lines
842 B
Python
from mlflow.entities import FileInfo
|
|
|
|
from tests.helper_functions import random_int, random_str
|
|
|
|
|
|
def _check(fi, path, is_dir, size_in_bytes):
|
|
assert isinstance(fi, FileInfo)
|
|
assert fi.path == path
|
|
assert fi.is_dir == is_dir
|
|
assert fi.file_size == size_in_bytes
|
|
|
|
|
|
def test_creation_and_hydration():
|
|
path = random_str(random_int(10, 50))
|
|
is_dir = random_int(10, 2500) % 2 == 0
|
|
size_in_bytes = random_int(1, 10000)
|
|
fi1 = FileInfo(path, is_dir, size_in_bytes)
|
|
_check(fi1, path, is_dir, size_in_bytes)
|
|
|
|
as_dict = {"path": path, "is_dir": is_dir, "file_size": size_in_bytes}
|
|
assert dict(fi1) == as_dict
|
|
|
|
proto = fi1.to_proto()
|
|
fi2 = FileInfo.from_proto(proto)
|
|
_check(fi2, path, is_dir, size_in_bytes)
|
|
|
|
fi3 = FileInfo.from_dictionary(as_dict)
|
|
_check(fi3, path, is_dir, size_in_bytes)
|