Function test_ka_gnn

tests/tool_tests/test_molecular_prediction.py:47–55

()

@pytest.mark.asyncio
@pytest.mark.mcp
async def test_ka_gnn():
    async with Client(MCP_URL) as client:

        async def call_tool(tool_name: str, **kwargs) -> Dict[str, Any]:
            res = await client.call_tool(tool_name, kwargs or {})
            return getattr(res, "data", None) or {}

        result = await call_tool("ka_gnn", smiles=ETHANOL, task="bbbp")
        assert result, f"ka_gnn returned unexpected: {result}"


@pytest.mark.asyncio