mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-13 13:36:15 +00:00
community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463)
Moved the following modules to new package langchain-community in a backwards compatible fashion: ``` mv langchain/langchain/adapters community/langchain_community mv langchain/langchain/callbacks community/langchain_community/callbacks mv langchain/langchain/chat_loaders community/langchain_community mv langchain/langchain/chat_models community/langchain_community mv langchain/langchain/document_loaders community/langchain_community mv langchain/langchain/docstore community/langchain_community mv langchain/langchain/document_transformers community/langchain_community mv langchain/langchain/embeddings community/langchain_community mv langchain/langchain/graphs community/langchain_community mv langchain/langchain/llms community/langchain_community mv langchain/langchain/memory/chat_message_histories community/langchain_community mv langchain/langchain/retrievers community/langchain_community mv langchain/langchain/storage community/langchain_community mv langchain/langchain/tools community/langchain_community mv langchain/langchain/utilities community/langchain_community mv langchain/langchain/vectorstores community/langchain_community mv langchain/langchain/agents/agent_toolkits community/langchain_community mv langchain/langchain/cache.py community/langchain_community mv langchain/langchain/adapters community/langchain_community mv langchain/langchain/callbacks community/langchain_community/callbacks mv langchain/langchain/chat_loaders community/langchain_community mv langchain/langchain/chat_models community/langchain_community mv langchain/langchain/document_loaders community/langchain_community mv langchain/langchain/docstore community/langchain_community mv langchain/langchain/document_transformers community/langchain_community mv langchain/langchain/embeddings community/langchain_community mv langchain/langchain/graphs community/langchain_community mv langchain/langchain/llms community/langchain_community mv langchain/langchain/memory/chat_message_histories community/langchain_community mv langchain/langchain/retrievers community/langchain_community mv langchain/langchain/storage community/langchain_community mv langchain/langchain/tools community/langchain_community mv langchain/langchain/utilities community/langchain_community mv langchain/langchain/vectorstores community/langchain_community mv langchain/langchain/agents/agent_toolkits community/langchain_community mv langchain/langchain/cache.py community/langchain_community ``` Moved the following to core ``` mv langchain/langchain/utils/json_schema.py core/langchain_core/utils mv langchain/langchain/utils/html.py core/langchain_core/utils mv langchain/langchain/utils/strings.py core/langchain_core/utils cat langchain/langchain/utils/env.py >> core/langchain_core/utils/env.py rm langchain/langchain/utils/env.py ``` See .scripts/community_split/script_integrations.sh for all changes
This commit is contained in:
@@ -0,0 +1,46 @@
|
||||
import unittest
|
||||
from typing import Any
|
||||
from unittest.mock import MagicMock, patch
|
||||
|
||||
from langchain_community.graphs import HugeGraph
|
||||
|
||||
|
||||
class TestHugeGraph(unittest.TestCase):
|
||||
def setUp(self) -> None:
|
||||
self.username = "test_user"
|
||||
self.password = "test_password"
|
||||
self.address = "test_address"
|
||||
self.graph = "test_hugegraph"
|
||||
self.port = 1234
|
||||
self.session_pool_size = 10
|
||||
|
||||
@patch("hugegraph.connection.PyHugeGraph")
|
||||
def test_init(self, mock_client: Any) -> None:
|
||||
mock_client.return_value = MagicMock()
|
||||
huge_graph = HugeGraph(
|
||||
self.username, self.password, self.address, self.port, self.graph
|
||||
)
|
||||
self.assertEqual(huge_graph.username, self.username)
|
||||
self.assertEqual(huge_graph.password, self.password)
|
||||
self.assertEqual(huge_graph.address, self.address)
|
||||
self.assertEqual(huge_graph.port, self.port)
|
||||
self.assertEqual(huge_graph.graph, self.graph)
|
||||
|
||||
@patch("hugegraph.connection.PyHugeGraph")
|
||||
def test_execute(self, mock_client: Any) -> None:
|
||||
mock_client.return_value = MagicMock()
|
||||
huge_graph = HugeGraph(
|
||||
self.username, self.password, self.address, self.port, self.graph
|
||||
)
|
||||
query = "g.V().limit(10)"
|
||||
result = huge_graph.query(query)
|
||||
self.assertIsInstance(result, MagicMock)
|
||||
|
||||
@patch("hugegraph.connection.PyHugeGraph")
|
||||
def test_refresh_schema(self, mock_client: Any) -> None:
|
||||
mock_client.return_value = MagicMock()
|
||||
huge_graph = HugeGraph(
|
||||
self.username, self.password, self.address, self.port, self.graph
|
||||
)
|
||||
huge_graph.refresh_schema()
|
||||
self.assertNotEqual(huge_graph.get_schema, "")
|
Reference in New Issue
Block a user