PyTestPilot

AI-powered test case generator for Python, built with Google Gemini.

Features

• Automatically generates professional pytest test cases using Gemini AI
• Extracts all functions from your Python file automatically
• Tests edge cases like None, zero, negative numbers, and empty strings
• Creates a test_yourfile.py file alongside your source file
• One-click button to run generated tests directly in the terminal
• Right-click any .py file to generate tests instantly

Setup

1. Get a free Gemini API key from aistudio.google.com
2. Open VSCode Settings (Ctrl+,)
3. Search for PyTestPilot
4. Paste your API key in the Gemini Api Key field

Usage

Option 1 — Command Palette

1. Open a Python file
2. Press Ctrl+Shift+P
3. Type PyTestPilot: Generate Tests and hit Enter

Option 2 — Right Click

1. Right-click any .py file in the Explorer or Editor
2. Select PyTestPilot: Generate Tests

Running Tests

1. Open the generated test_yourfile.py file
2. Click the ▶ play button in the top right of the editor toolbar
3. pytest runs automatically in the terminal

Requirements

• Python installed on your system
• pytest installed (pip install pytest)
• A free Gemini API key from aistudio.google.com

Author

Gunveer Singh

If you find PyTestPilot useful, consider leaving a ⭐ review on the marketplace!