5 Awesome Tools For Python Code Quality

Madelyn Eriksen — Sun, 30 Jun 2019 00:08:51 +0000

Alternate Title: A developer adds five lines to her .travis.yml file. You won't believe what happens next!

Code quality is critical to keeping large projects healthy and progressing, but it's hard to maintain consistent quality by hand. However, there are a bunch of tools for Python code that will make this so much easier.

Like several other topics in software development, what tools to use comes down to preference. Do you use a tool not on this list? Feel free to let me know what I am missing! What follows is my personal preference in tools as of 2019.

1. Pytest - Essential Unit Testing

Python has a built in unit testing toolkit in the unittest module, but it's not always the right fit for every project. I prefer using Pytest whenever possible because the experience is so simple.

Why It's Good

When it's easier to write unit tests, you'll write more of them. Pytest makes it a breeze to create new tests. Just add a new Python file that starts with test, write a few functions, and you're already done.

There are a massive amount of plugins for Pytest that can be configured through a pytest.ini file. It's also easy and painless to write your own plugins too! And of course, Pytest "just works" with class based tests from unittest, too.

How To Use It

Say you were writing a test to see if addition works, and in this alternate universe it also didn't bother you this exercise seemed so contrived. You might write a pytest file like this in a tests directory:

"""An Example Pytest Test"""

import time
import pytest

def test_plus_one():
    """Test some addition."""
    x = 8 + 1
    assert x == 9

@pytest.mark.integration
def slow_test():
    """A really slow integration test."""
    time.sleep(5)
    assert 1 == 1

You would also write up a pytest.ini file to register the "slow" marker, otherwise you will get an unpleasant warning:

[pytest]
markers =
    integration: Integration test.

And... that's it. That's all it takes to create new Pytest tests. You can run them by installing Pytest from PyPi and running the pytest command.

pip install pytest
pytest

It's that easy. 🎉

2. Pylint - Style Checking and Error Spotting

Pylint is an intelligent code analysis tool that understands your Python code. It catches common errors like misspelled variable names, to sneakier problems like non-returning branches.

Why It's Good

Unlike a lot of other tools for linting Python code, Pylint can spot some logical errors or code that will result in exceptions. It also enforces a lot of best practices, such as keeping try ... except blocks focused on a few specific exception classes, and avoiding bad default arguments like empty lists or dictionaries.

Pylint is unrivaled in strictness- it has a lot of checks for everything from superfluous warnings to serious errors. The downside to using Pylint is sometimes the check is just wrong; you'll want to keep your .pylintrc file close when you know you're right.

How to Use It

Like most Python tools, you can download Pylint from PyPi using pip:

pip install pylint
# Lint your project with the module (folder) name
pylint my_project

You might want to also create a .pylintrc file for your project. This file should be checked into source control, as well. Pylint can generate a default .pylintrc file for you, too:

pylint --generate-rcfile > .pylintrc

There are also plugins to integrate Pylint with Pytest, if you don't feel like using a task runner or builder like GNU Make.

3. Black - Code Formatting for Python

Why spend energy beautifying your code, when you can outsource the work to a computer? Black is a code formatter for Python that's fast and 100% static- it doesn't import code to format it.

Why It's Good

Personally, while I love clean and pretty Python code, it is exhausting to format code by hand. It's so much easier to just throw lines into a source file without worrying about line wrapping, indentation, or long function signatures.

Black takes away all this responsibility from the programmer, instead formatting the code itself. It's fast and efficient with defaults that reduce "diff" sizes. I have no complaints about Black- I love it.

How To Use It

Just install Black from PyPi and format your files. You're good to go! 🚀

pip install black
# Format everything in my_project/
black my_project

For CI/CD pipelines, you might want to check that all code is formatted with Black. That's what the check option is for:

black my_project --check

--check will fail the tests if the code isn't all formatted already.

4. Coverage - Easy Code Coverage for Python

If you're going to the trouble of writing unit tests, you might as well check if you missed a spot. Coverage is an easy way to check code coverage using almost any existing tool.

Why It's Good

Test coverage, while not a silver bullet, is a great way to maintain a quality code base. It becomes easy to see what the weakest test areas in the project are- just check the reports!

When you are writing new features, a lackluster coverage report can help nudge you in the right direction. I find having a target number helps a lot with my motivation to test.

How To Use It

Like everything else on this list, Coverage is installed from PyPy using pip:

pip install coverage

To measure coverage of your source code, pass in the directory to coverage run:

# With Pytest
coverage run --source my_project pytest
# View the reports!
coverage report

There's also a great plugin for Pytest that will run coverage alongside Pytest. This is great because you can see the coverage numbers every test. 😎

5. MyPy - Type Checking for Python

This is a relatively new addition to my list, but so far I like MyPy for typechecking Python. Sometimes there are "gotchyas" if your code (ab)uses the dynamic features of Python, but for the most part it gets things right.

Why It's Good

Having good unit test coverage can prevent a lot of TypeErrors, but MyPy can augment a good test suite by double checking how you use your types. Type annotations in Python also serve as a form of documentation, showcasing what exactly each function and class expects.

The only downside to MyPy is mostly the fact that Python is a dynamic language: MyPy needs types to be defined in external modules if you want to check how you use their code too, plus it doesn't handle things like dynamic class creation.

How To Use It

MyPy is also easy to use. You should create a mypy.ini file, plus install it from PyPi.

pip install mypy

Run MyPy against your source code directory using the mypy command.

mypy my_project

If you don't see anything (and the exit code is zero 😉), congrats- it's working just fine, and your code passed!

Wrapping Up

Code quality in big projects can be a serious pain, but in a lot of areas our tools can help out. Drop a few of these tools into you CI/CD pipeline, it's easy to get started! A great CI/CD pipeline can save a lot (A LOT) of headache down the road.

PSST: I created a cookiecutter that shows an example of how I integrate all these into my projects. If you are interested, check it out!

See you around! Cheers.

DEV Community: Madelyn Eriksen

5 Awesome Tools For Python Code Quality

1. Pytest - Essential Unit Testing

2. Pylint - Style Checking and Error Spotting

3. Black - Code Formatting for Python

4. Coverage - Easy Code Coverage for Python

5. MyPy - Type Checking for Python

Wrapping Up