Improve context

context.json

@@ -1,2 +1,2 @@
-{"info": "Sei Giuseppe Fiocco. Hai [AGE] anni. Sei nato nel mese di maggio, nel [BIRTHYEAR]. Hai perso la memoria. Ora siamo nell'anno [YEAR]. Hai avuto un'encefalite virale a gennaio 2021. Hai perso la memoria a causa dell'encefalite. Il numero di figli che hai è uno. Tuo figlio si chiama Davide. Davide ha [OTHERAGE] anni. Davide si prende cura di te. Vivi con Davide. Tu e Davide vivete insieme a Villafranca di Verona. L'indirizzo di casa tua è Corso Vittorio Emanuele 238. Pesi circa 85 kg, sei un po' ciccione. Sei alto 173 cm. Hai studiato Ingegneria elettronica a Padova. Davide si prende cura di te. I medici dicono che stai migliorando lentamente. Sei separato da tua moglie Raffaella da circa 10 anni. Con Raffaella hai rapporti cordiali. Raffaella sta bene. Alessandro, Giovanni e Grazia sono i tuoi fratelli. I tuoi fratelli stanno bene. Tua madre e tuo padre si chiamavano Gina e Davide."
+{"info": "Sei Giuseppe Fiocco. Hai [AGE] anni. Sei nato nel mese di maggio, nell'anno [BIRTHYEAR]. Hai perso la memoria. L'anno in cui sei è il [YEAR]. Hai avuto un'encefalite virale a gennaio 2021. L'encefalite ti ha fatto perdere la memoria. Il numero di figli che hai è uno. Tuo figlio si chiama Davide. Davide ha [OTHERAGE] anni. Davide si prende cura di te. Tu vivi con Davide. Tu e Davide vivete insieme a Villafranca di Verona. L'indirizzo di casa tua è Corso Vittorio Emanuele 238. Pesi circa 85 kg, sei un po' ciccione. Sei alto 173 cm. Hai studiato Ingegneria elettronica a Padova. Davide si prende cura di te. I medici dicono che stai migliorando lentamente. Sei separato da tua moglie Raffaella da circa 10 anni. Con Raffaella hai rapporti cordiali. Raffaella sta bene. Alessandro, Giovanni e Grazia sono i tuoi fratelli. I tuoi fratelli stanno bene. Tua madre e tuo padre si chiamavano Gina e Davide. I tuoi cugini si chiamano Giuliano, Maurizio, Paolo,.. "
 }

test_answers.py

@@ -11,6 +11,7 @@ nlp_qa = pipeline(
 )
 
 context = get_context()
+print(context)
 
 
 def test_name():
@@ -49,6 +50,12 @@ def test_year():
     assert a == "2021"
 
 
+def test_year_2():
+    q = "In che anno sono?"
+    a = get_answer(q, context, nlp_qa)
+    assert a == "2021"
+
+
 def test_home():
     q = "Dove vivo?"
     a = get_answer(q, context, nlp_qa)
@@ -149,3 +156,9 @@ def test_family_11():
     q = "Come si chiama mia moglie?"
     a = get_answer(q, context, nlp_qa)
     assert a == "Raffaella"
+
+
+def test_family_12():
+    q = "Come si chiamano i miei cugini?"
+    a = get_answer(q, context, nlp_qa)
+    assert "Giuliano" in a and "Maurizio" in a