Las pruebas en el software son un aspecto importante; pues permiten verificar que nuestro código se ejecuta en la manera que debería y que con cada cambio que hacemos no rompemos otras cosas.

Python provee el framework unittest que realiza pruebas unitarias de nuestro código y tiene varios métodos para realizar pruebas de código.

Hoy veremos cómo realizar nuestros primeros tests con el framework unittest de Python, que ya viene instalado por defecto.

Instalar Python

Si todavía no has instalado Python o no tienes las nociones básicas te invito a instalarlo en Windows, Linux Ubuntu o incluso en Android.

Documentación de unittest en Python

La documentación oficial se encuentra en el sitio oficial, ahí puedes indagar más sobre el tema. Aquí daré una breve explicación que sentará las bases de las pruebas unitarias en Python.

Escribir funciones

Vamos a escribir algunas funciones que realizan operaciones matemáticas; todo esto para ejemplificar y no confundir al lector con ejemplos complejos.

Tenemos cuatro operaciones que van a sumar, restar, dividir y multiplicar. Las 4 están en el siguiente archivo:

See the gist on github.

El archivo en sí mismo no está llamándolas, solamente las define. Ahora veamos cómo se pueden probar.

Definir tests unitarios con unittest

Para realizar las pruebas se requiere escribir una clase que extienda de unittest.TestCase. Dentro de esa clase definimos funciones que serán las encargadas de probar otras funciones; una función de prueba debe estar presente por cada función.

See the gist on github.

En el código fuente vemos que se utiliza assertEqual, esto recibe dos argumentos: el valor real o actual, y el valor esperado.

Los métodos setUp y tearDown son llamados antes de empezar cada test y después de terminar cada test, respectivamente.

Al final hacemos la comprobación de if __name__ == '__main__' para que, si el archivo es ejecutado directamente,los tests sean ejecutados, solo en caso de que el archivo sea ejecutado directamente, no importado.

Probar tests con python

Cuando ya definimos nuestros test, los ejecutamos como un archivo normal de Python así:

python probar.py

Actualmente existe el descubrimiento de tests y cosas de esas pero por ahora veremos lo básico. Recomiendo pasar la opción -v para que muestre más detalles de lo que está pasando:

python probar.py -v

La salida es como se ve a continuación:

Los 4 tests unitarios fueron pasados. Ahora voy a cambiar el método de la suma (para que devuelva a/b en lugar de a+b) y ejecutar las pruebas de nuevo:

Nota: olvidé pasar la opción -v pero es lo mismo, el test ha fallado y nos lo está indicando:

AssertionError: 0.5 != 3

Efectivamente, debido a que está devolviendo la división y no la suma los valores de esperado y actual no coinciden.

Otros métodos de assert

No solamente existe assertEqual, también existen los siguientes (he puesto solo algunos):

• assertNotEqual: lo contrario de assertEqual, asegurarse de que dos valores no son iguales.
• assertTrue: asegurar que un valor es True
• assertFalse: asegurar que un valor es False
• assertRaises: asegurar que se genera una excepción

Todo eso lo encuentras más detallado en los enlaces que he anexado.

Conclusión

No escribas pocos tests, escribe varios. No solamente escribas pruebas “buenas” sino también malas. Por ejemplo, haz tests para probar que se genera una excepción cuando pasas determinados valores.

Prueba todo tu código con distintos escenarios, entre más pruebas unitarias escribas menores errores tendrás en producción.

Aprender más sobre Python.