From 26fd86427743f9a1c518e2612902fbf5b4315683 Mon Sep 17 00:00:00 2001 From: sangjanai Date: Tue, 25 Feb 2025 09:11:20 +0700 Subject: [PATCH] chore: unit test --- engine/e2e-test/test_api_docker.py | 2 +- engine/test/components/test_huggingface_utils.cc | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/engine/e2e-test/test_api_docker.py b/engine/e2e-test/test_api_docker.py index 7276f2078..4757c1550 100644 --- a/engine/e2e-test/test_api_docker.py +++ b/engine/e2e-test/test_api_docker.py @@ -2,7 +2,7 @@ import requests from utils.test_runner import wait_for_websocket_download_success_event -repo_branches = ["tinyllama:1b-gguf"] +repo_branches = ["tinyllama:gguf"] class TestCortexsoModels: diff --git a/engine/test/components/test_huggingface_utils.cc b/engine/test/components/test_huggingface_utils.cc index 8377200e5..46701d0b0 100644 --- a/engine/test/components/test_huggingface_utils.cc +++ b/engine/test/components/test_huggingface_utils.cc @@ -10,8 +10,8 @@ TEST_F(HuggingFaceUtilTestSuite, TestGetModelRepositoryBranches) { EXPECT_GE(branches.value().size(), 3); EXPECT_EQ(branches.value()["main"].name, "main"); EXPECT_EQ(branches.value()["main"].ref, "refs/heads/main"); - EXPECT_EQ(branches.value()["1b-gguf"].name, "1b-gguf"); - EXPECT_EQ(branches.value()["1b-gguf"].ref, "refs/heads/1b-gguf"); + EXPECT_EQ(branches.value()["1b"].name, "1b"); + EXPECT_EQ(branches.value()["1b"].ref, "refs/heads/1b"); EXPECT_EQ(branches.value()["gguf"].name, "gguf"); EXPECT_EQ(branches.value()["gguf"].ref, "refs/heads/gguf"); }