36  Pruebas unitarias

import pandas as pd
import unittest

class PruebasDataFrame(unittest.TestCase):
    
    def setUp(self):
        # Inicializar el DataFrame en el método setUp para que se ejecute antes de cada prueba
        self.df = pd.read_parquet(self.f)
    
    def test_tamanio_dataframe(self):
        # Tamaño esperado del DataFrame
        tamanio_esperado = 52560
        
        # Verificar que el DataFrame tiene el tamaño esperado
        tamanio_actual = len(self.df)
        self.assertEqual(tamanio_actual, tamanio_esperado, 
                         f"El DataFrame tiene {tamanio_actual} elementos, se esperaba {tamanio_esperado}")

def run_tests(file_path):
    # Establecer la variable de clase para el archivo a probar
    PruebasDataFrame.f = file_path
    
    # Cargar y ejecutar el conjunto de pruebas
    suite = unittest.TestLoader().loadTestsFromTestCase(PruebasDataFrame)
    result = unittest.TextTestRunner().run(suite)
    
    # Devolver el resultado del test (True si pasó, False si falló)
    return result.wasSuccessful()

# Ruta del archivo Parquet
file_path = '../data/Temixco_2018_10Min.parquet'

# Ejecutar las pruebas
if run_tests(file_path):
    print("Todas las pruebas pasaron exitosamente.")
else:
    print("Algunas pruebas fallaron.")

.
----------------------------------------------------------------------
Ran 1 test in 0.047s

OK

Todas las pruebas pasaron exitosamente.