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

refining ProportionalReportingRatiosPersisterTest

Browse Source
This commit is contained in:
Frank Knoll
2024-07-11 22:54:29 +02:00
parent 58b08909a9
commit 53ac140e83
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/SymptomsCausedByVaccines/ProportionalReportingRatiosPersister.py
View File

@@ -3,11 +3,11 @@ from IOUtils import IOUtils
def saveProportionalReportingRatios(prrBySymptomByDrug, directory): def saveProportionalReportingRatios(prrBySymptomByDrug, directory):
shutil.rmtree(directory, ignore_errors = True) shutil.rmtree(directory, ignore_errors = True)
drugByFilename = {} filenameByDrug = {}
i = 0 i = 0
for drug, prrBySymptom in prrBySymptomByDrug.items(): for drug, prrBySymptom in prrBySymptomByDrug.items():
i += 1 i += 1
filename = f'{i}.json' filename = f'{i}.json'
drugByFilename[filename] = drug filenameByDrug[drug] = filename
IOUtils.saveDictAsJson(prrBySymptom, f'{directory}/{filename}') IOUtils.saveDictAsJson(prrBySymptom, f'{directory}/{filename}')
IOUtils.saveDictAsJson(drugByFilename, f'{directory}/drugByFilename.json') IOUtils.saveDictAsJson(filenameByDrug, f'{directory}/filenameByDrug.json')

4
src/SymptomsCausedByVaccines/ProportionalReportingRatiosPersisterTest.py
View File

@@ -26,8 +26,8 @@ class ProportionalReportingRatiosPersisterTest(unittest.TestCase):
ProportionalReportingRatiosPersisterTest.readJsonFile(f'{directory}/{drugFilename}'), ProportionalReportingRatiosPersisterTest.readJsonFile(f'{directory}/{drugFilename}'),
prrBySymptom) prrBySymptom)
self.assertDictEqual( self.assertDictEqual(
ProportionalReportingRatiosPersisterTest.readJsonFile(f'{directory}/drugByFilename.json'), ProportionalReportingRatiosPersisterTest.readJsonFile(f'{directory}/filenameByDrug.json'),
{ drugFilename: drug }) { drug: drugFilename })
@staticmethod @staticmethod
def readJsonFile(file): def readJsonFile(file):