Improve import test

tests/test_import.py

@@ -1,16 +1,33 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
+import os
+import sys
 
-import pytest
-from pytest import raises, approx
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
 
+import faster_llm
+from faster_llm.LLM.mcp import send_mcp_request
+from faster_llm.LLM.client import MCPClient
 
-def test():
-    assert 4 == 4
 
+def test_package_has_version():
+    assert isinstance(faster_llm.__version__, str)
 
-if __name__ == "__main__":
-    import os
 
-    basename = os.path.basename(__file__)
-    pytest.main([basename, "-s", "--tb=native"])
+def test_send_mcp_request_uses_client(monkeypatch):
+    captured = {}
+
+    def fake_send_request(self, method, params):
+        captured["url"] = self.server_url
+        captured["method"] = method
+        captured["params"] = params
+        return {"ok": True}
+
+    monkeypatch.setattr(MCPClient, "send_request", fake_send_request)
+
+    result = send_mcp_request("ping", {"x": 1}, "http://example.com")
+
+    assert result == {"ok": True}
+    assert captured == {
+        "url": "http://example.com",
+        "method": "ping",
+        "params": {"x": 1},
+    }