Comprobar con assert (tu primer paso hacia las pruebas)

En esta lección verás cómo usar assert para comprobaciones rápidas, y así asimilar la mentalidad básica de las pruebas. Está pensada para quien busca «tutorial de assert en Python» o «cómo escribir pruebas en Python».

assert condición, mensaje lanza un error solo cuando la condición es False. Es la idea básica del código de pruebas: comprobar automáticamente si el resultado de una función coincide con lo esperado. Los proyectos reales suelen usar una herramienta específica como pytest, pero la idea de fondo es la misma que la de este assert.

El código de ejemplo usa assert para confirmar que add(2, 3) debería ser 5, y comprueba a propósito una expectativa incorrecta —que debería ser 6—, capturando el AssertionError resultante con try...except y mostrando su mensaje. Fíjate en la diferencia: cuando la condición se cumple no pasa nada; cuando no, se lanza un error.

Un error habitual al empezar es no saber que Python tiene un modo de optimización que desactiva por completo los assert, algo importante si dependes de ellos en producción. Como pueden omitirse en ese modo, las comprobaciones que sí o sí deben ejecutarse en producción a veces es más seguro escribirlas como un if normal más una excepción.

Adquirir la costumbre de comparar el resultado esperado con el real es clave para detectar fallos pronto. Acostumbrarte a una comprobación rápida con assert cada vez que escribes una función hace que las correcciones y refactorizaciones posteriores den mucho menos miedo.