francois/HowBadIsMyBatch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

adding CreateDataFrameTest

Browse Source
This commit is contained in:
frankknoll
2022-01-28 23:37:37 +01:00
parent a40f3fe038
commit 8a496bc4d9
2 changed files with 70 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

71
HowBadIsMyBatch.ipynb
View File

@@ -44,7 +44,7 @@
" folder = dataDir + \"/\" + year + \"VAERSData/\"\n",
" return {\n",
" 'VAERSDATA': read_csv(folder + year + \"VAERSDATA.csv\", ['VAERS_ID', 'DIED', 'L_THREAT', 'DISABLE']),\n",
" 'VAERSVAX': read_csv(folder + year + \"VAERSVAX.csv\", ['VAERS_ID', 'VAX_TYPE', 'VAX_MANU', 'VAX_LOT'])\n",
" 'VAERSVAX': read_csv(folder + year + \"VAERSVAX.csv\", ['VAERS_ID', 'VAX_DOSE_SERIES', 'VAX_TYPE', 'VAX_MANU', 'VAX_LOT'])\n",
" }\n",
"\n",
" return _createDataFrame(\n",
@@ -85,12 +85,71 @@
{
"cell_type": "code",
"execution_count": null,
"id": "e14465d7",
"metadata": {},
"outputs": [],
"source": [
"from pandas.testing import assert_frame_equal\n",
"\n",
"class CreateDataFrameTest(unittest.TestCase):\n",
"\n",
" def test_createDataFrame(self):\n",
" # Given\n",
" vaersDescrs = [\n",
" {\n",
" 'VAERSDATA': self.createDataFrame(\n",
" [ 'DIED', 'L_THREAT', 'DISABLE'],\n",
" {\n",
" '0916600': ['Y', np.NaN, np.NaN],\n",
" '0916601': [np.NaN, np.NaN, 'Y']\n",
" }),\n",
" 'VAERSVAX': self.createDataFrame(\n",
" [ 'VAX_TYPE', 'VAX_MANU', 'VAX_LOT'],\n",
" {\n",
" '0916600': ['COVID19', 'MODERNA', '037K20A'],\n",
" '0916601': ['COVID19', 'MODERNA', '025L20A']\n",
" })\n",
" },\n",
" {\n",
" 'VAERSDATA': self.createDataFrame(\n",
" [ 'DIED', 'L_THREAT', 'DISABLE'],\n",
" {\n",
" '1996873': [np.NaN, np.NaN, np.NaN],\n",
" '1996874': [np.NaN, np.NaN, 'Y']\n",
" }),\n",
" 'VAERSVAX': self.createDataFrame(\n",
" [ 'VAX_TYPE', 'VAX_MANU', 'VAX_LOT'],\n",
" {\n",
" '1996873': ['HPV9', 'MERCK & CO. INC.', 'R017624'],\n",
" '1996874': ['COVID19', 'MODERNA', '025L20A']\n",
" })\n",
" }\n",
" ]\n",
" \n",
" # When\n",
" dataFrame = _createDataFrame(vaersDescrs, \"MODERNA\")\n",
" \n",
" # Then\n",
" dataFrameExpected = self.createDataFrame(\n",
" [ 'DIED', 'L_THREAT', 'DISABLE', 'VAX_TYPE', 'VAX_MANU', 'VAX_LOT'],\n",
" {\n",
" '0916600': ['Y', np.NaN, np.NaN, 'COVID19', 'MODERNA', '037K20A'],\n",
" '0916601': [np.NaN, np.NaN, 'Y', 'COVID19', 'MODERNA', '025L20A'],\n",
" '1996874': [np.NaN, np.NaN, 'Y', 'COVID19', 'MODERNA', '025L20A']\n",
" })\n",
" assert_frame_equal(dataFrame, dataFrameExpected, check_dtype = False)\n",
"\n",
" def createDataFrame(self, columns, data):\n",
" return pd.DataFrame.from_dict(data, columns = columns, orient = 'index')\n"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "e14465d7",
"metadata": {},
"outputs": [],
"source": [
"from pandas.testing import assert_frame_equal\n",
"\n",
"class BatchCodeTableTest(unittest.TestCase):\n",
"\n",
@@ -135,18 +194,18 @@
"\n",
" def _test_createBatchCodeTable(self, dataFrame):\n",
" # When\n",
" batchCodeTable=createBatchCodeTable(dataFrame)\n",
" batchCodeTable = createBatchCodeTable(dataFrame)\n",
"\n",
" # Then\n",
" batchCodeTableExpected=pd.DataFrame(\n",
" batchCodeTableExpected = pd.DataFrame(\n",
" data={\n",
" 'ADRs': [2, 1],\n",
" 'DEATHS': [0, 1],\n",
" 'DISABILITIES': [2, 0],\n",
" 'LIFE THREATENING ILLNESSES': [0, 0]\n",
" },\n",
" index=pd.MultiIndex.from_arrays([['025L20A', '037K20A']], names = ('VAX_LOT',)))\n",
" assert_frame_equal(batchCodeTable, batchCodeTableExpected, check_dtype=False)\n",
" index = pd.MultiIndex.from_arrays([['025L20A', '037K20A']], names = ('VAX_LOT',)))\n",
" assert_frame_equal(batchCodeTable, batchCodeTableExpected, check_dtype = False)\n",
"\n",
" def createDataFrame(self, columns, data):\n",
" return pd.DataFrame.from_dict(data, columns = columns, orient = 'index')\n"