Update context and tests

.vscode/settings.json

@@ -0,0 +1,7 @@
+{
+    "python.testing.pytestArgs": [
+        "."
+    ],
+    "python.testing.unittestEnabled": false,
+    "python.testing.pytestEnabled": true
+}

app.py

@@ -37,9 +37,9 @@ with st.spinner(
 with open("context.json") as f:
     context = json.load(f)
 
-st.title("La macchina che risponde alle tue domande personali")
+st.title("Le risposte alle tue domande personali")
 
-input = st.text_input("Fai una domanda e comparirà la risposta!")
+input = st.text_input("Scrivi una domanda e comparirà la risposta!")
 
 if input:
     try:

context.json

@@ -1,2 +1,2 @@
-{"info": "Sei Giuseppe, hai 69 anni e abiti e vivi a Villafranca di Verona con Davide in Corso Vittorio Emanuele 238. Ora siamo a novembre dell'anno 2021 e hai avuto un'encefalite virale a gennaio 2021. Davide è tuo figlio, ha 37 anni. A causa dell'encefalite hai perso la memoria. Tua moglie Raffaella abita a Verona. Raffaella sta bene. Con Raffaella hai rapporti cordiali. Alessandro, Giovanni e Grazia sono i tuoi fratelli. I tuoi fratelli stanno bene."
+{"info": "Sei Giuseppe, hai 69 anni e abiti e vivi a Villafranca di Verona con Davide in Corso Vittorio Emanuele 238. Hai perso la memoria. Ora siamo a novembre dell'anno 2021 e hai avuto un'encefalite virale a gennaio 2021. Hai perso la memoria a causa dell'encefalite. Hai i capelli grigi. Pesi circa 85 kg, sei un po' ciccione. Hai 10 dita delle mani e 10 dita dei piedi.  Davide è tuo figlio, ha 37 anni. I medici dicono che stai migliorando lentamente. Tua moglie Raffaella abita a Verona. Raffaella sta bene. Sei separato da Raffaella da circa 10 anni. Con Raffaella hai rapporti cordiali. Alessandro, Giovanni e Grazia sono i tuoi fratelli. I tuoi fratelli stanno bene."
 }

test_answers.py

@@ -24,12 +24,30 @@ def test_age():
     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)
@@ -58,3 +76,15 @@ 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"