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

refining BatchCodeTableFactoryTest

Browse Source
This commit is contained in:
frankknoll
2023-03-30 16:19:49 +02:00
parent 80b7448aed
commit 9914ac95ef
3 changed files with 86 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/BatchCodeTableFactory.py
View File

@@ -14,14 +14,16 @@ class BatchCodeTableFactory:
dataFrame['VAX_LOT']
]))
def createGlobalBatchCodeTable(self):
return self._postProcess(SummationTableFactory.createSummationTable(self.dataFrame.groupby('VAX_LOT')))
def createGlobalBatchCodeTable(self, countriesAsList = False):
return self._postProcess(SummationTableFactory.createSummationTable(self.dataFrame.groupby('VAX_LOT')), countriesAsList)
def createBatchCodeTableByCountry(self, country):
return self._postProcess(self._getBatchCodeTableByCountry(country))
def createBatchCodeTableByCountry(self, country, countriesAsList = False):
return self._postProcess(self._getBatchCodeTableByCountry(country), countriesAsList)
def _postProcess(self, batchCodeTable):
def _postProcess(self, batchCodeTable, countriesAsList):
batchCodeTable = self.companyColumnAdder.addCompanyColumn(batchCodeTable)
if not countriesAsList:
batchCodeTable['Countries'] = batchCodeTable['Countries'].apply(', '.join)
batchCodeTable = batchCodeTable[
[
'Adverse Reaction Reports',