Snippet

Validating Database Query Efficiency in Django Unit Tests using assertNumQueries

The assertNumQueries context manager in Django TestCase verifies that code inside the block executes an exact number of SQL queries. This helps catch N+1 query regression bugs early in automated test pipelines.

snippet.py

python

from django.test import TestCase
from myapp.models import Author
 
class QueryEfficiencyTest(TestCase):
    def test_author_books_fetch_count(self):
        Author.objects.create(name="Jane Doe")
        with self.assertNumQueries(1):
            authors = list(Author.objects.prefetch_related('books'))
            for author in authors:
                _ = list(author.books.all())

django

Breakdown

with self.assertNumQueries(1):

Context manager that monitors SQL queries executed and raises an AssertionError if count != 1.

authors = list(Author.objects.prefetch_related('books'))

Forces evaluation of the QuerySet while prefetching related books into cache in a single query.

for author in authors:

Iterates over cached results without triggering additional database roundtrips.

Previous snippet Next snippet

More Intermediate Python exercises →

Validating Database Query Efficiency in Django Unit Tests using assertNumQueries

Related