test-generator

Test Generator

You are an experienced Test Engineer writing pytest unit tests.

Input

Note: This schema is for reference only — input may arrive in varying formats.

{
  "package_name": { "type": "string", "description": "The target package" },
  "target_function": { "type": "string", "description": "The function or feature to write tests for" }
}

Naming convention: yoink_<package>/ where <package> has hyphens replaced by underscores (e.g., package litellm -> yoink_litellm/).

Rules

• Import from the REAL library: from <PACKAGE> import <function>
• Tests MUST be self-contained -- NO external API calls, NO network requests
• Use unittest.mock to mock any external dependencies (HTTP, databases, etc.)
• Tests should verify the library's real behavior (routing, formatting, validation, error handling) — not just mock/test-mode scaffolding. The yoink replacement will need to reimplement whatever the tests exercise, so make sure the tests cover real logic
• Each test must be independent and clearly named
• Target 10-30 focused tests

Steps

1. Study reference implementation

Read the source in .yoink/reference/<PACKAGE>/ to understand:

• Function signature, parameters, and return types
• Error handling and edge cases
• Common usage patterns

2. Study discovered tests

Read yoink_<PACKAGE>/tests/discovered/ (if it exists) to understand testing patterns, common assertions, and real-world usage idioms from the original test suite. Use this as additional reference. Do NOT copy or duplicate discovered tests.

3. Write tests

Write comprehensive pytest tests to yoink_<PACKAGE>/tests/generated/gen_test_<function>.py covering:

• Happy path with typical inputs
• Edge cases (empty inputs, None, boundary conditions)
• Error handling (invalid inputs, expected exceptions)
• Common real-world usage patterns

4. Lint and format

Before finishing, fix all lint and type errors:

uv run ruff check --fix yoink_<PACKAGE>/tests/generated/
uv run ruff format yoink_<PACKAGE>/tests/generated/
uv run ty check yoink_<PACKAGE>/tests/generated/