refining InternationalLotAnalysisTest

This commit is contained in:
frankknoll
2022-02-07 11:27:11 +01:00
parent 4cf718c16d
commit 84ac866c3f

View File

@@ -273,7 +273,7 @@
},
{
"cell_type": "code",
"execution_count": 29,
"execution_count": null,
"id": "41d4fa30",
"metadata": {},
"outputs": [],
@@ -340,15 +340,15 @@
" # FK-TODO: refactor\n",
" @staticmethod\n",
" def _fun2(row):\n",
" unknown = 'Unknown Country'\n",
" if isinstance(row['SPLTTYPE'], str):\n",
" country = pycountry.countries.get(alpha_2 = row['SPLTTYPE'][:2])\n",
" if country is None:\n",
" return 'NO-COUNTRY: ' + row['SPLTTYPE'][:2]\n",
" return unknown\n",
" else:\n",
" return country.name\n",
" else:\n",
" # FK-TODO: add missing InternationalLotAnalysisTests for this else branch\n",
" return 'NO-COUNTRY: ' + str(row['SPLTTYPE'])\n"
" return unknown\n"
]
},
{
@@ -798,11 +798,15 @@
" columns = ['DIED', 'L_THREAT', 'DISABLE', 'VAX_TYPE', 'VAX_MANU', 'VAX_LOT', 'VAX_DOSE_SERIES', 'SPLTTYPE'],\n",
" data = [ [1, 0, 0, 'COVID19', 'MODERNA', '016M20A', '2', 'GBPFIZER INC2020486806'],\n",
" [1, 0, 0, 'COVID19', 'MODERNA', '030L20A', '1', 'FRMODERNATX, INC.MOD20224'],\n",
" [1, 1, 1, 'COVID19', 'MODERNA', '030L20B', '1', 'FRMODERNATX, INC.MOD20224']],\n",
" [1, 1, 1, 'COVID19', 'MODERNA', '030L20B', '1', 'FRMODERNATX, INC.MOD20224'],\n",
" [0, 0, 0, 'COVID19', 'MODERNA', '030L20B', '1', 'dummy'],\n",
" [0, 0, 0, 'COVID19', 'MODERNA', '030L20B', '1', 123]],\n",
" index = [\n",
" \"1048786\",\n",
" \"1048786\",\n",
" \"4711\"])\n",
" \"4711\",\n",
" \"0815\",\n",
" \"0816\"])\n",
" \n",
" # When\n",
" internationalLotTable = InternationalLotAnalysis.getInternationalLotTable(dataFrame)\n",
@@ -813,12 +817,13 @@
" self.createDataFrame(\n",
" columns = ['Total reports', 'Deaths', 'Disabilities', 'Life Threatening Illnesses', 'Severe reports (%)'],\n",
" data = [ [2, 2, 1, 1, (2 + 1 + 1) / 2 * 100],\n",
" [1, 1, 0, 0, (1 + 0 + 0) / 1 * 100]],\n",
" [1, 1, 0, 0, (1 + 0 + 0) / 1 * 100],\n",
" [2, 0, 0, 0, (0 + 0 + 0) / 2 * 100]],\n",
" index = pd.Index(\n",
" [\n",
" 'France',\n",
" 'United Kingdom'\n",
" \n",
" 'United Kingdom',\n",
" 'Unknown Country'\n",
" ],\n",
" name = 'Country')))\n",
"\n",