Davide Fiocco commited on
Commit
5febb07
·
1 Parent(s): e2af350

Add context and tests

Browse files
Files changed (2) hide show
  1. context.json +1 -1
  2. test_answers.py +12 -0
context.json CHANGED
@@ -1,2 +1,2 @@
1
- {"info": "Sei Giuseppe, hai 69 anni. Hai perso la memoria. Sei nato in Veneto. 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. Hai perso la memoria a causa dell'encefalite. Hai i capelli grigi. Pesi circa 85 kg, sei un po' ciccione. Hai studiato ingegneria, sei intelligente ma molto smemorato. 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."
2
  }
 
1
+ {"info": "Sei Giuseppe. Hai 69 anni. Hai perso la memoria. Sei nato in Veneto. 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. Hai perso la memoria a causa dell'encefalite. Hai i capelli grigi. Pesi circa 85 kg, sei un po' ciccione. Sei alto 173 cm. Hai studiato Ingegneria elettronica a Padova, sei intelligente ma molto smemorato. 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."
2
  }
test_answers.py CHANGED
@@ -42,6 +42,18 @@ def test_history():
42
  assert "encefalite" in a
43
 
44
 
 
 
 
 
 
 
 
 
 
 
 
 
45
  def test_recovery():
46
  q = "Come va il mio recupero?"
47
  a = get_answer(q, context, nlp_qa)
 
42
  assert "encefalite" in a
43
 
44
 
45
+ def test_studies():
46
+ q = "Cosa ho studiato?"
47
+ a = get_answer(q, context, nlp_qa)
48
+ assert a == "Ingegneria elettronica"
49
+
50
+
51
+ def test_studies_2():
52
+ q = "Dove ho studiato?"
53
+ a = get_answer(q, context, nlp_qa)
54
+ assert a == "Padova"
55
+
56
+
57
  def test_recovery():
58
  q = "Come va il mio recupero?"
59
  a = get_answer(q, context, nlp_qa)