refining ProportionalReportingRatiosPersisterTest
This commit is contained in:
@@ -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')
|
||||
|
||||
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user