test_basic_cosine_similarity() — langchain Function Reference

Architecture documentation for the test_basic_cosine_similarity() function in test_utils.py from the langchain codebase.

Function python LangChainCore ApiManagement

Entity Profile

LangChainCore→ ApiManagement→ test_basic_cosine_similarity() — langchain Function Reference

Dependency Diagram

graph TD
  d1ee0c30_dafe_e168_937e_fd9caa758766["test_basic_cosine_similarity()"]
  43305d93_a58a_e7db_615b_b8a5a18284ab["TestCosineSimilarity"]
  d1ee0c30_dafe_e168_937e_fd9caa758766 -->|defined in| 43305d93_a58a_e7db_615b_b8a5a18284ab
  style d1ee0c30_dafe_e168_937e_fd9caa758766 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/core/tests/unit_tests/vectorstores/test_utils.py lines 16–23

    def test_basic_cosine_similarity(self) -> None:
        """Test basic cosine similarity calculation."""
        # Simple orthogonal vectors
        x: list[list[float]] = [[1, 0], [0, 1]]
        y: list[list[float]] = [[1, 0], [0, 1]]
        result = _cosine_similarity(x, y)
        expected = np.array([[1.0, 0.0], [0.0, 1.0]])
        np.testing.assert_array_almost_equal(result, expected)