Davide Fiocco commited on
Commit
94ea41f
·
1 Parent(s): 36576fd

Add today's date without regressions

Browse files
Files changed (3) hide show
  1. context.json +1 -1
  2. test_answers.py +6 -1
  3. utils.py +1 -0
context.json CHANGED
@@ -1,2 +1,2 @@
1
- {"info": "Sei Giuseppe Fiocco. Hai [AGE] anni. Sei nato nel mese di maggio, nell'anno [BIRTHYEAR]. Hai perso la memoria. I medici dicono che stai migliorando lentamente. 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 all'università di Padova. Sei separato da tua moglie Raffaella. 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,.. "
2
  }
 
1
+ {"info": "Sei Giuseppe Fiocco. Hai [AGE] anni. Sei nato nel mese di maggio, nell'anno [BIRTHYEAR]. Hai perso la memoria. I medici dicono che stai migliorando lentamente. L'anno in cui sei è il [YEAR]. Oggi é il giorno [TODAY]. Hai avuto un'encefalite virale a gennaio 2021. L'encefalite ti ha fatto perdere la memoria. Pesi circa 85 kg, sei un po' ciccione. Sei alto 173 cm. Hai studiato Ingegneria elettronica all'università di Padova. 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. Tua moglie si chiama Raffaella. Sei separato da Raffaella. Raffaella non vive con te. Tua moglie Raffaella vive a Verona. 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,.. "
2
  }
test_answers.py CHANGED
@@ -11,7 +11,6 @@ nlp_qa = pipeline(
11
  )
12
 
13
  context = get_context()
14
- print(context)
15
 
16
 
17
  def test_name():
@@ -159,6 +158,12 @@ def test_family_11():
159
 
160
 
161
  def test_family_12():
 
 
 
 
 
 
162
  q = "Come si chiamano i miei cugini?"
163
  a = get_answer(q, context, nlp_qa)
164
  assert "Giuliano" in a and "Maurizio" in a
 
11
  )
12
 
13
  context = get_context()
 
14
 
15
 
16
  def test_name():
 
158
 
159
 
160
  def test_family_12():
161
+ q = "Dove vive mia moglie?"
162
+ a = get_answer(q, context, nlp_qa)
163
+ assert a == "Verona"
164
+
165
+
166
+ def test_family_13():
167
  q = "Come si chiamano i miei cugini?"
168
  a = get_answer(q, context, nlp_qa)
169
  assert "Giuliano" in a and "Maurizio" in a
utils.py CHANGED
@@ -24,6 +24,7 @@ def get_context():
24
  context = (
25
  json.load(f)["info"]
26
  .replace("[YEAR]", str(now.year))
 
27
  .replace("[BIRTHYEAR]", str(BIRTHYEAR))
28
  .replace("[AGE]", str(now.year - BIRTHYEAR))
29
  .replace("[OTHERAGE]", str(now.year - OTHERBIRTHYEAR))
 
24
  context = (
25
  json.load(f)["info"]
26
  .replace("[YEAR]", str(now.year))
27
+ .replace("[TODAY]", f"{datetime.datetime.now():%d-%m-%Y}")
28
  .replace("[BIRTHYEAR]", str(BIRTHYEAR))
29
  .replace("[AGE]", str(now.year - BIRTHYEAR))
30
  .replace("[OTHERAGE]", str(now.year - OTHERBIRTHYEAR))