Skip to main content

python ut test code generator.

Project description

Python UT generator

This tools generate automatically Python pytest Unit test code.
This project uses ast module to generate.
Easy to make coverage test. And Easy to customize test code.


  • Generate unit test python file in tests package.
  • Generate pytest test function from each function.
  • Generate mock patch syntax code.
  • Generate argument syntax code to call.
  • If function has return value, create assert return.


Install pip

pip install pyutgenerator

Run tool.

Genarete test code

pyutgen "Input File Name"


python -m "Input File Name"

Sample input file

import os

def aaaaa():
    call and return
    return os.path.exists('')

Sample out put

import pytest
from unittest.mock import patch
from unittest.mock import MagicMock

from import pattern01

def test_aaaaa():
    # plan

    # do
            patch('') as m1:
        m1.return_value = None
        m1.exists = MagicMock(return_value=None)
        ret = pattern01.aaaaa()

        # check
        assert ret

For the future

  • For Code
    • Genarete various parameters for test.
    • Write return check value.
    • 'exception' check.
    • 'with' description mock.
    • Generate test data.
    • Assertion for method call for count, parameter, throw.
    • parameter type for str,list, obj ...
    • Simple code analyst report. like no None check or Parameter
    • For django
  • Customize parameter options or setting file.
    • Exclude function mock.
    • Source directory path.
    • tab space num.
  • Get test data from debug.
  • able to run default generated code and pass test.
  • Full coverage.
  • web ui for test.


not yet

not yet


This project is licensed under the MIT License - see the file for details


  • Hat tip to anyone whose code was used
  • Inspiration
  • etc

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

pyutgenerator-0.12.0.tar.gz (11.2 kB view hashes)

Uploaded Source

Built Distribution

pyutgenerator-0.12.0-py3-none-any.whl (13.1 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page