import os | |
import pytest | |
from src.whisper_audio_extractor import record_audio, transcribe_audio, AUDIO_PATH | |
def test_record_audio(): | |
""" | |
Verifica que la función de grabación crea un archivo de audio con un tamaño válido. | |
""" | |
# Ejecuta la grabación con una duración de prueba corta | |
record_audio(duration=2) # Graba por 2 segundos para el test | |
# Comprueba si el archivo de audio existe | |
assert os.path.exists(AUDIO_PATH), "El archivo de audio no fue creado." | |
# Comprueba que el archivo no esté vacío | |
assert os.path.getsize(AUDIO_PATH) > 0, "El archivo de audio está vacío." | |
def test_transcribe_audio(): | |
""" | |
Verifica que la función de transcripción devuelve texto. | |
""" | |
# Ejecuta la transcripción del audio grabado | |
transcription = transcribe_audio() | |
# Asegura que se obtuvo texto | |
assert isinstance(transcription, str) and len(transcription) > 0, "La transcripción está vacía o no es texto." | |
if __name__ == "__main__": | |
pytest.main() | |