Function test_mock_llm_stream_matches_sync_text

tests/test_providers.py:135–142

()

@pytest.mark.unit
def test_mock_llm_stream_matches_sync_text() -> None:
    llm = MockLLM()
    prompt = "hello " * 20
    sync = llm.generate(prompt, system="sys")
    chunks = list(llm.stream_generate(prompt, system="sys"))

    assert chunks
    assert "".join(chunks) == sync.text


@pytest.mark.unit