From 3fb2c0299121b9738e70f9279b679d838181466c Mon Sep 17 00:00:00 2001
From: Frank Knoll <Knoll_Frank@web.de>
Date: Thu, 11 Jul 2024 10:41:10 +0200
Subject: [PATCH] starting ProportionalReportingRatiosPersisterTest

---
 ...roportionalReportingRatiosPersisterTest.py | 30 +++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 src/SymptomsCausedByVaccines/ProportionalReportingRatiosPersisterTest.py

diff --git a/src/SymptomsCausedByVaccines/ProportionalReportingRatiosPersisterTest.py b/src/SymptomsCausedByVaccines/ProportionalReportingRatiosPersisterTest.py
new file mode 100644
index 00000000000..cc94d01795b
--- /dev/null
+++ b/src/SymptomsCausedByVaccines/ProportionalReportingRatiosPersisterTest.py
@@ -0,0 +1,30 @@
+import unittest
+from pandas.testing import assert_series_equal
+from TestHelper import TestHelper
+import pandas as pd
+import simplejson as json
+from SymptomsCausedByVaccines.ProportionalReportingRatiosPersister import saveProportionalReportingRatios
+
+
+class ProportionalReportingRatiosPersisterTest(unittest.TestCase):
+
+    def test_saveProportionalReportingRatios(self):
+        # Given
+        prrBySymptom = {'Sepsis ': 366.3084378480811}
+        prrBySymptomByDrug = TestHelper.createSeries(
+                indexName = 'DRUG',
+                data = {
+                    '"GENERIC COLD ACID" ': prrBySymptom
+                })
+        
+        # When
+        saveProportionalReportingRatios(prrBySymptomByDrug, 'src/tmp/vaccines')
+        
+        # Then
+        data = ProportionalReportingRatiosPersisterTest.readJsonFile('src/tmp/vaccines/"GENERIC COLD ACID" .json')
+        self.assertDictEqual(data, prrBySymptom)
+
+    @staticmethod
+    def readJsonFile(file):
+        with open(file) as fp:
+            return json.load(fp)