langchain Architecture Docs
Home / File/ test_base.py — langchain Source File

test_base.py — langchain Source File

Architecture documentation for test_base.py, a python file in the langchain codebase. 9 imports, 0 dependents.

File python CoreAbstractions Serialization 9 imports 9 functions 6 classes

Entity Profile

Dependency Diagram

graph LR
  4404310a_b16c_e9e6_1de2_90e5ea6f3ce3["test_base.py"]
  cfe2bde5_180e_e3b0_df2b_55b3ebaca8e7["collections.abc"]
  4404310a_b16c_e9e6_1de2_90e5ea6f3ce3 --> cfe2bde5_180e_e3b0_df2b_55b3ebaca8e7
  8e2034b7_ceb8_963f_29fc_2ea6b50ef9b3["typing"]
  4404310a_b16c_e9e6_1de2_90e5ea6f3ce3 --> 8e2034b7_ceb8_963f_29fc_2ea6b50ef9b3
  120e2591_3e15_b895_72b6_cb26195e40a6["pytest"]
  4404310a_b16c_e9e6_1de2_90e5ea6f3ce3 --> 120e2591_3e15_b895_72b6_cb26195e40a6
  91721f45_4909_e489_8c1f_084f8bd87145["typing_extensions"]
  4404310a_b16c_e9e6_1de2_90e5ea6f3ce3 --> 91721f45_4909_e489_8c1f_084f8bd87145
  f3bc7443_c889_119d_0744_aacc3620d8d2["langchain_core.callbacks"]
  4404310a_b16c_e9e6_1de2_90e5ea6f3ce3 --> f3bc7443_c889_119d_0744_aacc3620d8d2
  ba43b74d_3099_7e1c_aac3_cf594720469e["langchain_core.language_models"]
  4404310a_b16c_e9e6_1de2_90e5ea6f3ce3 --> ba43b74d_3099_7e1c_aac3_cf594720469e
  ac2a9b92_4484_491e_1b48_ec85e71e1d58["langchain_core.outputs"]
  4404310a_b16c_e9e6_1de2_90e5ea6f3ce3 --> ac2a9b92_4484_491e_1b48_ec85e71e1d58
  0dd80fba_b72d_91a8_6113_3bb8b0a7af05["langchain_core.tracers.context"]
  4404310a_b16c_e9e6_1de2_90e5ea6f3ce3 --> 0dd80fba_b72d_91a8_6113_3bb8b0a7af05
  c2b939d8_ea65_23ed_f553_c7331ab00ffa["tests.unit_tests.fake.callbacks"]
  4404310a_b16c_e9e6_1de2_90e5ea6f3ce3 --> c2b939d8_ea65_23ed_f553_c7331ab00ffa
  style 4404310a_b16c_e9e6_1de2_90e5ea6f3ce3 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

from collections.abc import AsyncIterator, Iterator
from typing import Any

import pytest
from typing_extensions import override

from langchain_core.callbacks import (
    AsyncCallbackManagerForLLMRun,
    CallbackManagerForLLMRun,
)
from langchain_core.language_models import (
    LLM,
    BaseLLM,
    FakeListLLM,
)
from langchain_core.outputs import Generation, GenerationChunk, LLMResult
from langchain_core.tracers.context import collect_runs
from tests.unit_tests.fake.callbacks import (
    BaseFakeCallbackHandler,
    FakeAsyncCallbackHandler,
    FakeCallbackHandler,
)


def test_batch() -> None:
    llm = FakeListLLM(responses=["foo"] * 3)
    output = llm.batch(["foo", "bar", "foo"])
    assert output == ["foo"] * 3

    output = llm.batch(["foo", "bar", "foo"], config={"max_concurrency": 2})
    assert output == ["foo"] * 3


async def test_abatch() -> None:
    llm = FakeListLLM(responses=["foo"] * 3)
    output = await llm.abatch(["foo", "bar", "foo"])
    assert output == ["foo"] * 3

    output = await llm.abatch(["foo", "bar", "foo"], config={"max_concurrency": 2})
    assert output == ["foo"] * 3


def test_batch_size() -> None:
    llm = FakeListLLM(responses=["foo"] * 3)
    with collect_runs() as cb:
        llm.batch(["foo", "bar", "foo"], {"callbacks": [cb]})
        assert all((r.extra or {}).get("batch_size") == 3 for r in cb.traced_runs)
        assert len(cb.traced_runs) == 3
    llm = FakeListLLM(responses=["foo"])
    with collect_runs() as cb:
        llm.batch(["foo"], {"callbacks": [cb]})
        assert all((r.extra or {}).get("batch_size") == 1 for r in cb.traced_runs)
        assert len(cb.traced_runs) == 1

    llm = FakeListLLM(responses=["foo"])
    with collect_runs() as cb:
        llm.invoke("foo")
        assert len(cb.traced_runs) == 1
        assert (cb.traced_runs[0].extra or {}).get("batch_size") == 1

// ... (220 more lines)

Domain

Subdomains

Functions

Classes

Dependencies

  • collections.abc
  • langchain_core.callbacks
  • langchain_core.language_models
  • langchain_core.outputs
  • langchain_core.tracers.context
  • pytest
  • tests.unit_tests.fake.callbacks
  • typing
  • typing_extensions

Source

Frequently Asked Questions

What does test_base.py do?
test_base.py is a source file in the langchain codebase, written in python. It belongs to the CoreAbstractions domain, Serialization subdomain.
What functions are defined in test_base.py?
test_base.py defines 9 function(s): test_abatch, test_astream_fallback_to_ainvoke, test_astream_implementation_fallback_to_stream, test_astream_implementation_uses_astream, test_async_batch_size, test_batch, test_batch_size, test_error_callback, test_get_ls_params.
What does test_base.py depend on?
test_base.py imports 9 module(s): collections.abc, langchain_core.callbacks, langchain_core.language_models, langchain_core.outputs, langchain_core.tracers.context, pytest, tests.unit_tests.fake.callbacks, typing, and 1 more.
Where is test_base.py in the architecture?
test_base.py is located at libs/core/tests/unit_tests/language_models/llms/test_base.py (domain: CoreAbstractions, subdomain: Serialization, directory: libs/core/tests/unit_tests/language_models/llms).

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free