test_multiple_vectors() — langchain Function Reference

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

Function python LangChainCore ApiManagement

Entity Profile

LangChainCore→ ApiManagement→ test_multiple_vectors() — langchain Function Reference

Dependency Diagram

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

Relationship Graph

Source Code

libs/core/tests/unit_tests/vectorstores/test_utils.py lines 48–60

    def test_multiple_vectors(self) -> None:
        """Test cosine similarity with multiple vectors."""
        x: list[list[float]] = [[1, 0], [0, 1], [1, 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],
                [1 / math.sqrt(2), 1 / math.sqrt(2)],
            ]
        )
        np.testing.assert_array_almost_equal(result, expected)