Testing

from rdk import init

init(mode="test")

from rdk import init

init(enabled=False)  # init() returns None, no instrumentation occurs

# conftest.py
import pytest
from rdk import init, shutdown

@pytest.fixture(autouse=True)
def rdk():
    init(mode="test")
    yield
    shutdown()

from rdk.testing import capture_traces
from rdk import observe
from anthropic import Anthropic

@observe()
def chat(message: str) -> str:
    client = Anthropic()
    response = client.messages.create(
        model="claude-sonnet-4-6",
        max_tokens=256,
        messages=[{"role": "user", "content": message}],
    )
    return response.content[0].text

def test_chat_creates_span():
    with capture_traces() as spans:
        chat("Hello")

    assert len(spans) == 1
    assert spans[0].name == "anthropic.messages.create"
    assert spans[0].metadata["provider"] == "anthropic"

def test_token_usage_captured():
    with capture_traces() as spans:
        chat("Hello")

    llm_span = spans[0]
    assert llm_span.token_usage is not None
    assert llm_span.token_usage.total_tokens > 0

from rdk import init
from rdk.testing import InMemoryTransport

def test_with_memory_transport():
    transport = InMemoryTransport()
    client = init(mode="test")
    client.set_transport(transport)

    chat("Hello")
    client.flush()

    assert transport.sent_count == 1
    assert len(transport.spans) == 1

# pytest.ini or pyproject.toml
[pytest]
env =
    RDK_MODE=test

RDK_MODE=test pytest