Pruebas unitarias con unittest

Las pruebas unitarias son un método para probar unidades individuales de código y garantizar que funcionen como se espera. 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.

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

El módulo unittest está incluido en 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 una subclase 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()

Para escribir un caso de prueba, defina un método en su clase de prueba. 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

Ejecute las pruebas mediante 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

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á dentro de b.
• assertRaises(exception): Comprueba si se produce una excepción.

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

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

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

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.

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 comunes disponibles en el framework unittest.
• Prácticas recomendadas para organizar los archivos de prueba y gestionar los casos límite.

¡Excelente trabajo! Pasemos al siguiente tema.