parsing(new_zealand): fix age parsing for Unknown age

globaldothealth · Oct 15, 2022 · 22429aa · 22429aa
1 parent 965f4d9
commit 22429aa
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/ingestion/functions/parsing/new_zealand/new_zealand.py b/ingestion/functions/parsing/new_zealand/new_zealand.py
@@ -63,7 +63,7 @@ def convert_demographics(entry):
     If age is listed as 90+, setting age range as between 90 and 120.
     '''
     demo = {}
-    if (age := entry[_AGE]) != "NA":
+    if (age := entry[_AGE]) not in ["NA", "Unknown"]:
         if '+' in age:
             start = int(age.rstrip('+'))
             demo["ageRange"] = {"start": start, "end": 120}

diff --git a/ingestion/functions/parsing/new_zealand/new_zealand_test.py b/ingestion/functions/parsing/new_zealand/new_zealand_test.py
@@ -83,6 +83,7 @@ def _c(location, gender, age_start, age_end, confirmed_date, travel=False, notes
     _c(_Bay_of_Plenty, "Female", 90, 120, "05/05/2021Z", True, "Case imported from abroad."),
     _c(_West_Coast, "Female", 20, 29, "05/06/2021Z", False),
     _c(_West_Coast, None, None, None, "07/01/2021Z", False),
+    _c(_West_Coast, None, None, None, "07/01/2021Z", False),
 ]
 
 

diff --git a/ingestion/functions/parsing/new_zealand/sample_data.csv b/ingestion/functions/parsing/new_zealand/sample_data.csv
@@ -10,3 +10,4 @@ Report Date,Case Status,Sex,Age group,DHB,Overseas travel,Historical
 2021-05-05,Confirmed,Female,90+,Bay of Plenty,Yes,
 2021-05-06,Confirmed,Female,20 to 29,West Coast,No,
 2021-07-01,Confirmed,Unknown,NA,West Coast,No,
+2021-07-01,Confirmed,Unknown,Unknown,West Coast,No,