refining ProportionalReportingRatiosPersisterTest

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

View File

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

View File

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