refining ProportionalReportingRatiosPersisterTest

src/SymptomsCausedByVaccines/ProportionalReportingRatiosPersister.py

@@ -1,7 +1,12 @@
 import shutil
 from IOUtils import IOUtils
 
-def saveProportionalReportingRatios(prrByVaccineBySymptom, directory):
+def saveProportionalReportingRatios(prrBySymptomByDrug, directory):
     shutil.rmtree(directory, ignore_errors = True)
-    for symptom, prrByVaccine in prrByVaccineBySymptom.items():
-        IOUtils.saveDictAsJson(prrByVaccine, f'{directory}/{symptom}.json')
+    drugByFilename = {}
+    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')    

src/SymptomsCausedByVaccines/ProportionalReportingRatiosPersisterTest.py

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