adding DataFrameFilterTest.test_filterByFlu()

This commit is contained in:
frankknoll
2022-02-16 16:53:53 +01:00
parent 4167f25fcb
commit c16183fc97

View File

@@ -147,6 +147,9 @@
" def filterByCovid19(self, dataFrame):\n",
" return dataFrame[self._isCovid19(dataFrame)]\n",
"\n",
" def filterByFlu(self, dataFrame):\n",
" return dataFrame[self._isFlu(dataFrame)]\n",
"\n",
" def filterByCountry(self, dataFrame, country, countryColumnName):\n",
" return dataFrame[dataFrame[countryColumnName] == country]\n",
"\n",
@@ -156,6 +159,9 @@
" def _isCovid19(self, dataFrame):\n",
" return dataFrame[\"VAX_TYPE\"] == \"COVID19\"\n",
"\n",
" def _isFlu(self, dataFrame):\n",
" return dataFrame[\"VAX_TYPE\"].str.startswith(\"FLU\")\n",
"\n",
" def _isManufacturer(self, dataFrame, manufacturer):\n",
" return dataFrame[\"VAX_MANU\"] == manufacturer if manufacturer is not None else True\n",
"\n",
@@ -530,6 +536,80 @@
"\n",
"class DataFrameFilterTest(unittest.TestCase):\n",
"\n",
" def test_filterByFlu(self):\n",
" # Given\n",
" dataFrame = TestHelper.createDataFrame(\n",
" columns = ['DIED', 'L_THREAT', 'DISABLE', 'VAX_TYPE', 'VAX_MANU', 'VAX_LOT', 'VAX_DOSE_SERIES'],\n",
" data = [ [0, 0, 0, 'FLU(H1N1)', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLU3', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLU4', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLUA3', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLUA4', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLUC3', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLUC4', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLUN(H1N1)', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLUN3', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLUN4', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLUR3', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLUR4', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLUX', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLUX(H1N1)', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 1, 'COVID19', 'MODERNA', '025L20A', '1']],\n",
" index = [\n",
" \"801410\",\n",
" \"801411\",\n",
" \"801412\",\n",
" \"801413\",\n",
" \"801414\",\n",
" \"801415\",\n",
" \"801416\",\n",
" \"801417\",\n",
" \"801418\",\n",
" \"801419\",\n",
" \"801420\",\n",
" \"801421\",\n",
" \"801422\",\n",
" \"801423\",\n",
" \"801424\"])\n",
" dataFrameFilter = DataFrameFilter()\n",
" \n",
" # When\n",
" dataFrameActual = dataFrameFilter.filterByFlu(dataFrame)\n",
" \n",
" # Then\n",
" dataFrameExpected = TestHelper.createDataFrame(\n",
" columns = ['DIED', 'L_THREAT', 'DISABLE', 'VAX_TYPE', 'VAX_MANU', 'VAX_LOT', 'VAX_DOSE_SERIES'],\n",
" data = [ [0, 0, 0, 'FLU(H1N1)', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLU3', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLU4', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLUA3', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLUA4', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLUC3', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLUC4', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLUN(H1N1)', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLUN3', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLUN4', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLUR3', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLUR4', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLUX', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1'],\n",
" [0, 0, 0, 'FLUX(H1N1)', 'GLAXOSMITHKLINE BIOLOGICALS', '5R3J5', '1']],\n",
" index = [\n",
" \"801410\",\n",
" \"801411\",\n",
" \"801412\",\n",
" \"801413\",\n",
" \"801414\",\n",
" \"801415\",\n",
" \"801416\",\n",
" \"801417\",\n",
" \"801418\",\n",
" \"801419\",\n",
" \"801420\",\n",
" \"801421\",\n",
" \"801422\",\n",
" \"801423\"])\n",
" assert_frame_equal(dataFrameActual, dataFrameExpected, check_dtype = False)\n",
"\n",
" def test_filterByCovid19_filterBy(self):\n",
" # Given\n",
" dataFrame = VaersDescr2DataFrameConverter.createDataFrameFromDescrs(\n",
@@ -1438,6 +1518,28 @@
"\n",
"sns.boxplot(x = \"CONCENTRATION\", y = \"Total Number of Adverse Reaction Reports\", data = batchCodeTable, order = order)"
]
},
{
"cell_type": "markdown",
"id": "259b6474",
"metadata": {},
"source": [
"### COVID-19 Vaccines vs. Flu Vaccines"
]
},
{
"cell_type": "markdown",
"id": "fa5c8480",
"metadata": {},
"source": [
"see https://www.bitchute.com/video/4HlIyBmOEJeY/ and https://www.bitchute.com/video/8wJYP2NpGwN2/"
]
},
{
"cell_type": "markdown",
"id": "37ccbee6",
"metadata": {},
"source": []
}
],
"metadata": {