test_single_vector_vs_multiple() — langchain Function Reference

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

Function python LangChainCore MessageInterface

Entity Profile

LangChainCore→ MessageInterface→ test_single_vector_vs_multiple() — langchain Function Reference

Dependency Diagram

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

Relationship Graph

Source Code

libs/core/tests/unit_tests/vectorstores/test_utils.py lines 136–151

    def test_single_vector_vs_multiple(self) -> None:
        """Test single vector against multiple vectors."""
        x: list[list[float]] = [[1, 1]]
        y: list[list[float]] = [[1, 0], [0, 1], [1, 1], [-1, -1]]
        result = _cosine_similarity(x, y)
        expected = np.array(
            [
                [
                    1 / math.sqrt(2),  # cos(45°)
                    1 / math.sqrt(2),  # cos(45°)
                    1.0,  # cos(0°)
                    -1.0,  # cos(180°)
                ]
            ]
        )
        np.testing.assert_array_almost_equal(result, expected)