From ab322aa2841a72b08176d3263ea2ff3dd77a00ba Mon Sep 17 00:00:00 2001 From: RerankerGuo <121015044+RerankerGuo@users.noreply.github.com> Date: Wed, 5 Aug 2026 13:48:38 +0800 Subject: [PATCH] test(embedder): isolate Ark factory tests from optional SDK --- tests/embedders/test_ark.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/embedders/test_ark.py b/tests/embedders/test_ark.py index 08c23aae6..991d173bd 100644 --- a/tests/embedders/test_ark.py +++ b/tests/embedders/test_ark.py @@ -7,8 +7,9 @@ class TestEmbedderFactory(unittest.TestCase): + @patch.object(ArkEmbedder, "__init__", return_value=None) @patch.object(ArkEmbedder, "embed") - def test_embed_single_text(self, mock_embed): + def test_embed_single_text(self, mock_embed, _mock_init): """Test embedding a single text.""" mock_embed.return_value = [[0.1, 0.2, 0.3, 0.4, 0.5, 0.6]] @@ -30,8 +31,9 @@ def test_embed_single_text(self, mock_embed): mock_embed.assert_called_once_with([text]) self.assertEqual(len(result[0]), 6) + @patch.object(ArkEmbedder, "__init__", return_value=None) @patch.object(ArkEmbedder, "embed") - def test_embed_batch_text(self, mock_embed): + def test_embed_batch_text(self, mock_embed, _mock_init): """Test embedding multiple texts at once.""" mock_embed.return_value = [ [0.1, 0.2, 0.3, 0.4, 0.5, 0.6],