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-12 07:44:29 +02:00
parent dabacf6199
commit 2cc3b29ed3
2 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/SymptomsCausedByVaccines/ProportionalReportingRatiosPersister.py
View File

@@ -7,7 +7,6 @@ def saveProportionalReportingRatios(prrBySymptomByDrug, directory):
i = 0
for drug, prrBySymptom in prrBySymptomByDrug.items():
i += 1
filename = f'{i}.json'
filenameByDrug[drug] = filename
IOUtils.saveDictAsJson(prrBySymptom, f'{directory}/{filename}')
filenameByDrug[drug] = f'{i}'
IOUtils.saveDictAsJson(prrBySymptom, f'{directory}/{i}.json')
return filenameByDrug

4
src/SymptomsCausedByVaccines/ProportionalReportingRatiosPersisterTest.py
View File

@@ -21,9 +21,9 @@ class ProportionalReportingRatiosPersisterTest(unittest.TestCase):
filenameByDrug = saveProportionalReportingRatios(prrBySymptomByDrug, directory)
# Then
drugFilename = '1.json'
drugFilename = '1'
self.assertDictEqual(
ProportionalReportingRatiosPersisterTest.readJsonFile(f'{directory}/{drugFilename}'),
ProportionalReportingRatiosPersisterTest.readJsonFile(f'{directory}/{drugFilename}.json'),
prrBySymptom)
self.assertDictEqual(filenameByDrug, { drug: drugFilename })