{"slug": "how-to-generate-unit-tests-from-a-function-using-ai", "title": "How to Generate Unit Tests from a Function Using AI", "summary": "A developer shared a prompt engineering technique for generating comprehensive unit tests from a function using AI. The method involves providing context about the function, specifying edge cases, and asking for a test plan before implementation. An audit prompt helps identify gaps in the generated tests.", "body_md": "You have a function, you need tests, and you're staring at a blank file. AI can generate a solid first draft of your unit test suite in under a minute — but only if you prompt it correctly. This walkthrough shows you exactly how to do it, step by step, with copy-paste prompts and real output patterns you can adapt immediately.\n\nMost engineers start with something like `\"write tests for this function\"`\n\nand paste their code. The output is usually technically correct but shallow: happy-path only, no edge cases, no error conditions, no boundary values. It looks like coverage but isn't.\n\nThe fix isn't a fancier AI model — it's a better prompt structure.\n\nBefore pasting code, tell the AI what the function is supposed to do, what it promises to callers, and what test framework you're using. One short paragraph of context doubles the quality of the output.\n\n```\nI have a TypeScript function that validates a user-submitted email address\nbefore it's stored to the database. It should return true for valid emails\nand false for invalid ones. I'm using Vitest with the `describe/it/expect`\npattern. No mocking is needed — the function is pure.\n\nHere is the function:\n\n[PASTE FUNCTION HERE]\n\nGenerate a comprehensive unit test suite. Cover: the happy path, boundary\ncases (empty string, null, very long strings), malformed inputs\n(missing @, multiple @, special characters), and any edge cases implied\nby the implementation itself.\n```\n\nThat last line — *\"edge cases implied by the implementation itself\"* — is the key instruction. It forces the model to read the actual logic rather than generating generic tests.\n\nFor non-trivial functions, add one extra step: ask for the test plan first, review it, then ask for the implementation. This costs 30 seconds and catches gaps before they get baked into code.\n\n```\nBefore writing any test code, list every test case you plan to cover for\nthe function above. Group them by: (1) valid inputs, (2) invalid inputs,\n(3) boundary/edge cases, (4) error conditions. Use plain English, one\nline per case. I'll confirm the list before you write any code.\n```\n\nScan the list. You'll almost always spot a case the model missed — and more importantly, a case *you* would have missed too. Add it back in plain English before you proceed.\n\nOnce the plan is approved, run the generation prompt:\n\n```\nNow write the full Vitest test suite based on the test plan above.\nRequirements:\n- One `describe` block named after the function\n- One `it` block per test case, with a descriptive string that reads like a\n  sentence (\"returns false when the input is an empty string\")\n- No external dependencies or mocks unless the function requires them\n- If any test case requires a fixture or factory, define it at the top of\n  the describe block\n- Add a one-line comment above any test that is non-obvious\n```\n\nThe descriptive `it`\n\nstrings matter. They're your documentation. When a test fails in CI at 2am, a string like `\"returns false when the input is an empty string\"`\n\ntells you exactly what broke without opening the file.\n\nDon't ship the AI output directly. Run this audit prompt against the generated suite:\n\n```\nReview the test suite you just wrote. Identify any of the following that\nare missing or weakly covered:\n1. Return value correctness (not just truthiness)\n2. Side effects the function might have\n3. Thrown exceptions or error types\n4. Performance-sensitive paths (very large inputs)\n5. Any assumption in the code that isn't tested\n\nFor each gap, either add a test case or explain why it's intentionally\nexcluded.\n```\n\nThis is where the real value compounds. The model will often find its own blind spots when asked directly — things like checking the *type* of a return value, not just its truthiness, or confirming an exception has the right message, not just that it was thrown.\n\nThis pattern is one of the ones I've packaged into *The AI Leverage Playbook: 50 Prompts & Workflows for Engineers* — but the version above is enough to get value on its own.\n\nIf the audit reveals many similar inputs that differ only in data, ask the model to refactor to a parametrized format:\n\n```\nRefactor any test cases that share the same assertion logic but differ\nonly in input data into a `it.each` table. Keep non-repetitive tests as\nindividual `it` blocks.\n```\n\nThis cuts file length dramatically and makes it obvious when you need to add another row versus a whole new test.\n\nWorking through all five steps for a typical utility function takes about 10 minutes. The result is a test suite that would have taken 45–60 minutes to write by hand — and usually has better edge case coverage than the manual version.\n\nIf your codebase already has tests, paste one existing test file alongside your prompt and add: `\"Match the style, naming conventions, and structure of the example test file below exactly.\"`\n\nThe AI will mirror your patterns, which means the output requires almost no style cleanup before review.\n\nI break down one workflow like this every week in The AI Leverage Weekly — practical, no fluff, free. It's the fastest way to build a repeatable AI-assisted engineering practice. Subscribe: [https://theaileverageweekly.beehiiv.com/subscribe?utm_source=devto&utm_medium=article&utm_campaign=long_w7](https://theaileverageweekly.beehiiv.com/subscribe?utm_source=devto&utm_medium=article&utm_campaign=long_w7)", "url": "https://wpnews.pro/news/how-to-generate-unit-tests-from-a-function-using-ai", "canonical_source": "https://dev.to/leveragenotes/how-to-generate-unit-tests-from-a-function-using-ai-173m", "published_at": "2026-06-22 09:01:20+00:00", "updated_at": "2026-06-22 09:09:20.638502+00:00", "lang": "en", "topics": ["artificial-intelligence", "developer-tools"], "entities": ["Vitest", "TypeScript", "The AI Leverage Playbook"], "alternates": {"html": "https://wpnews.pro/news/how-to-generate-unit-tests-from-a-function-using-ai", "markdown": "https://wpnews.pro/news/how-to-generate-unit-tests-from-a-function-using-ai.md", "text": "https://wpnews.pro/news/how-to-generate-unit-tests-from-a-function-using-ai.txt", "jsonld": "https://wpnews.pro/news/how-to-generate-unit-tests-from-a-function-using-ai.jsonld"}}