refactoring
This commit is contained in:
@@ -106,21 +106,21 @@
|
||||
" # Given\n",
|
||||
" dataFrame = createDataFrame(\"test/VAERS\", \"MODERNA\")\n",
|
||||
" display(\"dataFrame:\", dataFrame)\n",
|
||||
" batchCodeTable = createBatchCodeTable(dataFrame)\n",
|
||||
"\n",
|
||||
" # When\n",
|
||||
" batchCodeTable = createBatchCodeTable(dataFrame)\n",
|
||||
" display(\"batchCodeTable:\", batchCodeTable)\n",
|
||||
"\n",
|
||||
" # Then\n",
|
||||
" batchCodeTableExpected = pd.DataFrame(\n",
|
||||
" {\n",
|
||||
" 'ADRs': [2, 1],\n",
|
||||
" 'DEATHS': [0, 1],\n",
|
||||
" 'DISABILITIES': [2, 0],\n",
|
||||
" 'LIFE THREATENING ILLNESSES': [0.0, 0.0]\n",
|
||||
" 'LIFE THREATENING ILLNESSES': [0, 0]\n",
|
||||
" },\n",
|
||||
" index = pd.MultiIndex.from_arrays([['025L20A', '037K20A']], names = ('VAX_LOT',)))\n",
|
||||
" display(\"batchCodeTable:\", batchCodeTable)\n",
|
||||
" display(\"batchCodeTableExpected:\", batchCodeTableExpected)\n",
|
||||
"\n",
|
||||
" # Then\n",
|
||||
" assert_frame_equal(batchCodeTable, batchCodeTableExpected, check_dtype = False)\n"
|
||||
]
|
||||
},
|
||||
|
||||
Reference in New Issue
Block a user