export: support float ages, convert to integers

Rarely ages are in float, this is more precision than necessary, so they are converted to integers. This fixes a ValueError encountered on float ages
globaldothealth · Oct 15, 2022 · 965f4d9 · 965f4d9
1 parent 93c04ce
commit 965f4d9
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/data-serving/scripts/export-data/test_convert_age.csv b/data-serving/scripts/export-data/test_convert_age.csv
@@ -7,3 +7,4 @@ demographics.ageRange.start,demographics.ageRange.end,demographics.ageBuckets
 0,0,[]
 ,,["0"]
 ,,"[""26-30"",""31-35""]"
+1.5,1.5,[]
diff --git a/data-serving/scripts/export-data/test_transform.py b/data-serving/scripts/export-data/test_transform.py
@@ -195,7 +195,7 @@ def convert_age_data():
     "row,expected_age",
     zip(convert_age_data(), [
         (20, 30), (21, 25), None, (21, 25),
-        (20, 30), (0, 0), (0, 0), (26, 35)
+        (20, 30), (0, 0), (0, 0), (26, 35), (1, 5),
     ])
 )
 def test_convert_age(row, expected_age, age_buckets):

diff --git a/data-serving/scripts/export-data/transform.py b/data-serving/scripts/export-data/transform.py
@@ -266,8 +266,8 @@ def convert_age(row: dict[str, Any], buckets: list[dict[str, Any]]) -> Optional[
     ):
         return None
 
-    age_start = int(row["demographics.ageRange.start"])
-    age_end = int(row["demographics.ageRange.end"])
+    age_start = int(float(row["demographics.ageRange.start"]))
+    age_end = int(float(row["demographics.ageRange.end"]))
     age_window = age_end - age_start + 1
 
     if age_window >= MINIMUM_AGE_WINDOW: