wangtinghu/agentUniverse
1
0
Fork 0
mirror of https://github.com/agentuniverse-ai/agentUniverse.git synced 2026-02-09 01:59:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #487 from ApricityXX/1029-3

Browse Source 
refactor: 增加测试断言
This commit is contained in:
Jerry Z H
2025-10-31 14:35:35 +08:00
committed by GitHub
parent b7f8ebb1c9 61be278b25
commit 7685821272
2 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_agentuniverse/unit/agent/action/knowledge/embedding/test_azureopenai_embedding.py
View File

@@ -41,6 +41,10 @@ class EmbeddingTest(unittest.TestCase):
langchain_embedding = self.embedding.as_langchain()
res = langchain_embedding.embed_documents(texts=["hello world"])
print(res)
self.assertIsInstance(res, list)
self.assertEqual(len(res), 1)
self.assertIsInstance(res[0], list)
self.assertEqual(len(res[0]), 1536)
if __name__ == '__main__':
unittest.main()

3
tests/test_agentuniverse/unit/agent/action/knowledge/embedding/test_embedding.py
View File

@@ -22,6 +22,9 @@ class EmbeddingTest(unittest.TestCase):
def test_get_embeddings(self) -> None:
res = self.embedding.get_embeddings(texts=["hello world"])
print(res)
self.assertIsNotNone(res)
self.assertIsInstance(res, list)
self.assertGreater(len(res), 0)
def test_async_get_embeddings(self) -> None:
res = asyncio.run(self.embedding.async_get_embeddings(texts=["hello world"]))