torbjorn/my-pal-mcp-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed tests

Browse Source
This commit is contained in:
Fahad
2025-06-12 21:00:53 +04:00
parent 3aedb16101
commit 22093bbf18
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_providers.py
View File

@@ -85,7 +85,7 @@ class TestGeminiProvider:
assert capabilities.provider == ProviderType.GOOGLE assert capabilities.provider == ProviderType.GOOGLE
assert capabilities.model_name == "gemini-2.5-flash-preview-05-20" assert capabilities.model_name == "gemini-2.5-flash-preview-05-20"
assert capabilities.max_tokens == 1_048_576 assert capabilities.max_tokens == 1_048_576
assert not capabilities.supports_extended_thinking assert capabilities.supports_extended_thinking
def test_get_capabilities_pro_model(self): def test_get_capabilities_pro_model(self):
"""Test getting capabilities for Pro model with thinking support""" """Test getting capabilities for Pro model with thinking support"""

2
tests/test_thinking_modes.py
View File

@@ -23,6 +23,7 @@ def setup_test_env():
class TestThinkingModes: class TestThinkingModes:
"""Test thinking modes across all tools""" """Test thinking modes across all tools"""
@patch("config.DEFAULT_THINKING_MODE_THINKDEEP", "high")
def test_default_thinking_modes(self): def test_default_thinking_modes(self):
"""Test that tools have correct default thinking modes""" """Test that tools have correct default thinking modes"""
tools = [ tools = [
@@ -169,6 +170,7 @@ class TestThinkingModes:
@pytest.mark.asyncio @pytest.mark.asyncio
@patch("tools.base.BaseTool.get_model_provider") @patch("tools.base.BaseTool.get_model_provider")
@patch("config.DEFAULT_THINKING_MODE_THINKDEEP", "high")
async def test_thinking_mode_max(self, mock_get_provider): async def test_thinking_mode_max(self, mock_get_provider):
"""Test max thinking mode (default for thinkdeep)""" """Test max thinking mode (default for thinkdeep)"""
mock_provider = create_mock_provider() mock_provider = create_mock_provider()