test_answers.py

from utils import get_answer
from transformers import pipeline
import json

nlp_qa = pipeline(
    "question-answering",
    model="mrm8488/bert-italian-finedtuned-squadv1-it-alfa",
    tokenizer="mrm8488/bert-italian-finedtuned-squadv1-it-alfa",
)

with open("context.json") as f:
    context = json.load(f)["info"]


def test_name():
    q = "Come mi chiamo?"
    a = get_answer(q, context, nlp_qa)
    assert a == "Giuseppe"


def test_age():
    q = "Quanti anni ho?"
    a = get_answer(q, context, nlp_qa)
    assert a == "69"


def test_year():
    q = "In che anno siamo?"
    a = get_answer(q, context, nlp_qa)
    assert a == "2021"


def test_home():
    q = "Dove vivo?"
    a = get_answer(q, context, nlp_qa)
    assert a == "Villafranca di Verona"


def test_history():
    q = "Cosa mi è successo?"
    a = get_answer(q, context, nlp_qa)
    assert "encefalite" in a


def test_recovery():
    q = "Come va il mio recupero?"
    a = get_answer(q, context, nlp_qa)
    assert "migliorando" in a


def test_family():
    q = "Con chi vivo?"
    a = get_answer(q, context, nlp_qa)
    assert a == "Davide"


def test_family_2():
    q = "Come si chiama mio figlio?"
    a = get_answer(q, context, nlp_qa)
    assert a == "Davide"


def test_family_3():
    q = "Quanti anni ha mio figlio?"
    a = get_answer(q, context, nlp_qa)
    assert a == "37"


def test_family_4():
    q = "Come sta Raffaella?"
    a = get_answer(q, context, nlp_qa)
    assert a == "Raffaella sta bene"


def test_family_5():
    q = "In che rapporti sono con Raffaella?"
    a = get_answer(q, context, nlp_qa)
    assert a == "cordiali"


def test_family_6():
    q = "Chi sono i miei fratelli?"
    a = get_answer(q, context, nlp_qa)
    assert a == "Alessandro, Giovanni e Grazia"


def test_family_7():
    q = "Come stanno i miei fratelli?"
    a = get_answer(q, context, nlp_qa)
    assert a == "I tuoi fratelli stanno bene"