From 53ac140e83abb339feb9e40f14a20176cc09e2b5 Mon Sep 17 00:00:00 2001
From: Frank Knoll <Knoll_Frank@web.de>
Date: Thu, 11 Jul 2024 22:54:29 +0200
Subject: [PATCH] refining ProportionalReportingRatiosPersisterTest

---
 .../ProportionalReportingRatiosPersister.py                 | 6 +++---
 .../ProportionalReportingRatiosPersisterTest.py             | 4 ++--
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/SymptomsCausedByVaccines/ProportionalReportingRatiosPersister.py b/src/SymptomsCausedByVaccines/ProportionalReportingRatiosPersister.py
index 8fd8b910634..4dcc1b5d98e 100644
--- a/src/SymptomsCausedByVaccines/ProportionalReportingRatiosPersister.py
+++ b/src/SymptomsCausedByVaccines/ProportionalReportingRatiosPersister.py
@@ -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')    
diff --git a/src/SymptomsCausedByVaccines/ProportionalReportingRatiosPersisterTest.py b/src/SymptomsCausedByVaccines/ProportionalReportingRatiosPersisterTest.py
index 18fc00dd39d..5f412d384f3 100644
--- a/src/SymptomsCausedByVaccines/ProportionalReportingRatiosPersisterTest.py
+++ b/src/SymptomsCausedByVaccines/ProportionalReportingRatiosPersisterTest.py
@@ -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):