tests: rework tests and fixes bugs along the way

This commit is contained in:
Mathieu Virbel
2023-08-01 16:05:48 +02:00
parent bc55cfdea3
commit 1f8e4200fd
9 changed files with 126 additions and 54 deletions

View File

@@ -0,0 +1,45 @@
import pytest
from unittest.mock import patch
@pytest.mark.asyncio
async def test_basic_process(event_loop):
# goal is to start the server, and send rtc audio to it
# validate the events received
from reflector.tools.process import process_audio_file
from reflector.settings import settings
from reflector.llm.base import LLM
from pathlib import Path
# use an LLM test backend
settings.LLM_BACKEND = "test"
class LLMTest(LLM):
async def _generate(self, prompt: str, **kwargs) -> str:
return {
"title": "TITLE",
"summary": "SUMMARY",
}
LLM.register("test", LLMTest)
# event callback
marks = {
"transcript": 0,
"topic": 0,
# "summary": 0,
}
async def event_callback(event, data):
print(f"{event}: {data}")
marks[event] += 1
# invoke the process and capture events
path = Path(__file__).parent / "records" / "test_mathieu_hello.wav"
await process_audio_file(path.as_posix(), event_callback)
print(marks)
# validate the events
assert marks["transcript"] == 5
assert marks["topic"] == 4
# assert marks["summary"] == 1