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 11:24:05 +02:00
parent 3fb2c02991
commit f1f38c32dc
2 changed files with 20 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/SymptomsCausedByVaccines/ProportionalReportingRatiosPersister.py
View File

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

17
src/SymptomsCausedByVaccines/ProportionalReportingRatiosPersisterTest.py
View File

@@ -10,19 +10,24 @@ class ProportionalReportingRatiosPersisterTest(unittest.TestCase):
def test_saveProportionalReportingRatios(self): def test_saveProportionalReportingRatios(self):
# Given # Given
drug = '"GENERIC COLD ACID" '
prrBySymptom = { 'Sepsis ': 366.3084378480811 } prrBySymptom = { 'Sepsis ': 366.3084378480811 }
prrBySymptomByDrug = TestHelper.createSeries( prrBySymptomByDrug = TestHelper.createSeries(
indexName = 'DRUG', indexName = 'DRUG',
data = { data = { drug: prrBySymptom })
'"GENERIC COLD ACID" ': prrBySymptom directory = 'src/tmp/vaccines'
})
# When # When
saveProportionalReportingRatios(prrBySymptomByDrug, 'src/tmp/vaccines') saveProportionalReportingRatios(prrBySymptomByDrug, directory)
# Then # Then
data = ProportionalReportingRatiosPersisterTest.readJsonFile('src/tmp/vaccines/"GENERIC COLD ACID" .json') drugFilename = '1.json'
self.assertDictEqual(data, prrBySymptom) self.assertDictEqual(
ProportionalReportingRatiosPersisterTest.readJsonFile(f'{directory}/{drugFilename}'),
prrBySymptom)
self.assertDictEqual(
ProportionalReportingRatiosPersisterTest.readJsonFile(f'{directory}/drugByFilename.json'),
{ drugFilename: drug })
@staticmethod @staticmethod
def readJsonFile(file): def readJsonFile(file):