NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-07-13 15:14:50 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: fix function tool UT

Browse Source 
PiperOrigin-RevId: 764567867
This commit is contained in:
Xiang (Sean) Zhou 2025-05-28 22:29:01 -07:00 committed by Copybara-Service
parent a0714b8afc
commit b92e068155
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/unittests/tools/test_function_tool.py
View File

@ -104,14 +104,16 @@ def test_init():
assert tool.func == function_for_testing_with_no_args
def test_function_returning_none():
@pytest.mark.asyncio
async def test_function_returning_none():
"""Test that the function returns with None actually returning None."""
tool = FunctionTool(function_returning_none)
result = await tool.run_async(args={}, tool_context=MagicMock())
assert result is None
def test_function_returning_empty_dict():
@pytest.mark.asyncio
async def test_function_returning_empty_dict():
"""Test that the function returns with empty dict actually returning empty dict."""
tool = FunctionTool(function_returning_empty_dict)
result = await tool.run_async(args={}, tool_context=MagicMock())