hf-qa-demo / tests /qa_engine /test_qa_engine.py
KonradSzafer's picture
pydantic class fix
0121498
raw
history blame
276 Bytes
import os
from qa_engine import Config, QAEngine
def test_qa_engine():
config = Config()
qa_engine = QAEngine(config=config)
response = qa_engine.get_response(
question='What is the capital of Poland?',
)
assert response.get_answer() == 'Warsaw'