File size: 1,025 Bytes
d4757ae |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
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()
|