test_python_code_splitter() — langchain Function Reference

Architecture documentation for the test_python_code_splitter() function in test_text_splitters.py from the langchain codebase.

Function python LangChainCore Runnables

Entity Profile

LangChainCore→ Runnables→ test_python_code_splitter() — langchain Function Reference

Dependency Diagram

graph TD
  a546b605_7018_ae03_f015_9a0ebf547be0["test_python_code_splitter()"]
  6d6b8ad4_1cfe_fbb0_e58e_76a50487c135["test_text_splitters.py"]
  a546b605_7018_ae03_f015_9a0ebf547be0 -->|defined in| 6d6b8ad4_1cfe_fbb0_e58e_76a50487c135
  style a546b605_7018_ae03_f015_9a0ebf547be0 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/text-splitters/tests/unit_tests/test_text_splitters.py lines 630–650

def test_python_code_splitter() -> None:
    splitter = RecursiveCharacterTextSplitter.from_language(
        Language.PYTHON, chunk_size=CHUNK_SIZE, chunk_overlap=0
    )
    code = """
def hello_world():
    print("Hello, World!")

# Call the function
hello_world()
    """
    chunks = splitter.split_text(code)
    assert chunks == [
        "def",
        "hello_world():",
        'print("Hello,',
        'World!")',
        "# Call the",
        "function",
        "hello_world()",
    ]