Pruebas unitarias con unittest

Las pruebas unitarias son un método para probar unidades individuales de código y comprobar que funcionan según lo esperado. El módulo unittest de Python es una biblioteca integrada para crear y ejecutar pruebas unitarias.

En esta lección, aprenderá a crear y ejecutar pruebas unitarias mediante el framework unittest.

¿Qué son las pruebas unitarias?

Las pruebas unitarias consisten en probar funciones o métodos individuales de forma aislada para comprobar que se comportan correctamente. Ayudan a detectar bugs en una etapa temprana del proceso de desarrollo y garantizan la fiabilidad del código.

Configuración de unittest

El módulo unittest forma parte de la biblioteca estándar de Python, por lo que no necesita instalar nada. Para usarlo, importe el módulo y cree casos de prueba mediante subclases de unittest.TestCase.

# Setting up unittest
import unittest

class TestExample(unittest.TestCase):
    def test_addition(self):
        self.assertEqual(1 + 1, 2)  # Passes

if __name__ == '__main__':
    unittest.main()

Escritura de casos de prueba

Para escribir un caso de prueba, defina un método en su clase de pruebas. Cada método debe comenzar por test_ para que el framework unittest pueda reconocerlo como un caso de prueba.

# Writing test cases
class TestMathOperations(unittest.TestCase):
    def test_multiplication(self):
        self.assertEqual(2 * 3, 6)  # Passes

Ejecución de pruebas

Ejecute las pruebas ejecutando el script que contiene sus casos de prueba. El framework unittest detectará y ejecutará automáticamente todos los métodos de prueba.

# Running tests
# Save this file and run it using:
# python test_script.py

Aserciones habituales

El módulo unittest proporciona varios métodos de aserción:

• assertEqual(a, b): Comprueba si a es igual a b.
• assertTrue(x): Comprueba si x es verdadero.
• assertIn(a, b): Comprueba si a está en b.
• assertRaises(exception): Comprueba si se produce una excepción.

Pruebas de casos límite

Es importante probar casos límite, como entradas vacías, números grandes o datos no válidos, para garantizar la solidez de su código.

# Testing edge cases
class TestEdgeCases(unittest.TestCase):
    def test_empty_input(self):
        self.assertEqual(len([]), 0)  # Passes

Organizar archivos de prueba

En proyectos grandes, organice los casos de prueba en una carpeta independiente, normalmente llamada tests/, y asigne a los archivos de prueba nombres con el prefijo test_.

Errores habituales con unittest

Estos son algunos errores que debe evitar:

• No nombrar los métodos de prueba con el prefijo test_.
• Olvidar gestionar los casos límite en los casos de prueba.
• No ejecutar las pruebas con regularidad durante el desarrollo.

¿Qué hemos aprendido?

En esta lección, ha aprendido:

• Qué son las pruebas unitarias y por qué son importantes.
• Cómo usar el módulo unittest para escribir y ejecutar casos de prueba.
• Las aserciones habituales disponibles en el marco de trabajo unittest.
• Prácticas recomendadas para organizar archivos de prueba y gestionar casos límite.

¡Excelente trabajo! Pasemos al siguiente tema.