From ec16658925d734fdaf453c7bf1567116a8475f95 Mon Sep 17 00:00:00 2001 From: gcoinstash-cmd Date: Sun, 13 Sep 2026 22:15:54 -0700 Subject: [PATCH] test(client): add unit test suite for model name tag whitespace trimming and colon splitting --- tests/test_model_name_trimming.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) create mode 100644 tests/test_model_name_trimming.py diff --git a/tests/test_model_name_trimming.py b/tests/test_model_name_trimming.py new file mode 100644 index 00000000..4d3fd57a --- /dev/null +++ b/tests/test_model_name_trimming.py @@ -0,0 +1,13 @@ +import pytest + +def parse_model_tag(model: str) -> tuple[str, str]: + cleaned = model.strip() + if ":" in cleaned: + parts = cleaned.split(":", 1) + return parts[0].strip(), parts[1].strip() + return cleaned, "latest" + +def test_parse_model_tag(): + assert parse_model_tag("llama3") == ("llama3", "latest") + assert parse_model_tag("mistral:7b") == ("mistral", "7b") + assert parse_model_tag(" qwen2.5:14b ") == ("qwen2.5", "14b")