Skip to content

Commit

Permalink
tests(mm): update tests to reflect using UUID for key
Browse files Browse the repository at this point in the history
  • Loading branch information
psychedelicious committed Mar 3, 2024
1 parent 554d175 commit 2f372d9
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions tests/app/services/model_install/test_model_install.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
"""

import platform
import uuid
from pathlib import Path

import pytest
Expand Down Expand Up @@ -30,9 +31,8 @@ def test_registration(mm2_installer: ModelInstallServiceBase, embedding_file: Pa
matches = store.search_by_attr(model_name="test_embedding")
assert len(matches) == 0
key = mm2_installer.register_path(embedding_file)
assert key is not None
assert key != "<NOKEY>"
assert len(key) == 32
# Not raising here is sufficient - key should be UUIDv4
uuid.UUID(key, version=4)


def test_registration_meta(mm2_installer: ModelInstallServiceBase, embedding_file: Path) -> None:
Expand Down

0 comments on commit 2f372d9

Please sign in to comment.