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 i = 0
for drug, prrBySymptom in prrBySymptomByDrug.items(): for drug, prrBySymptom in prrBySymptomByDrug.items():
i += 1 i += 1
filename = f'{i}.json' filenameByDrug[drug] = f'{i}'
filenameByDrug[drug] = filename IOUtils.saveDictAsJson(prrBySymptom, f'{directory}/{i}.json')
IOUtils.saveDictAsJson(prrBySymptom, f'{directory}/{filename}')
return filenameByDrug return filenameByDrug

4
src/SymptomsCausedByVaccines/ProportionalReportingRatiosPersisterTest.py
View File

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