From 6790ddd12e9dc824dc724c6a4585d3cbdf0707f1 Mon Sep 17 00:00:00 2001
From: frankknoll <Knoll_Frank@web.de>
Date: Tue, 10 Oct 2023 08:12:45 +0200
Subject: [PATCH] refining AnalyzerTest

---
 src/SymptomsCausedByVaccines/Analyzer.py     |  6 ++++
 src/SymptomsCausedByVaccines/AnalyzerTest.py | 38 ++++++++++++++++++++
 2 files changed, 44 insertions(+)

diff --git a/src/SymptomsCausedByVaccines/Analyzer.py b/src/SymptomsCausedByVaccines/Analyzer.py
index 8953fd1abe2..bfbd0dd932e 100644
--- a/src/SymptomsCausedByVaccines/Analyzer.py
+++ b/src/SymptomsCausedByVaccines/Analyzer.py
@@ -10,3 +10,9 @@ class Analyzer:
 
     def getVaccinesForSymptom(self, symptom):
         return self.symptomByVaccine[symptom]
+
+    def getVaccines(self):
+        return list(self.symptomByVaccine.index)
+
+    def getSymptoms(self):
+        return list(self.symptomByVaccine.columns)
diff --git a/src/SymptomsCausedByVaccines/AnalyzerTest.py b/src/SymptomsCausedByVaccines/AnalyzerTest.py
index 18c46ab852f..d55a0e113e7 100644
--- a/src/SymptomsCausedByVaccines/AnalyzerTest.py
+++ b/src/SymptomsCausedByVaccines/AnalyzerTest.py
@@ -58,3 +58,41 @@ class AnalyzerTest(unittest.TestCase):
                     'ADEN': 0.3
                     },
                 indexName = 'VAX_TYPE'))
+
+    def test_getVaccines(self):
+        # Given
+        symptomByVaccine = TestHelper.createDataFrame(
+                columns = ['11-beta-hydroxylase deficiency'],
+                data = [  [0.6],
+                          [0.3]],
+                index = pd.Index(
+                    name = 'VAX_TYPE',
+                    data = [
+                        '6VAX-F',
+                        'ADEN'
+                        ]))
+        
+        analyzer = Analyzer(symptomByVaccine)
+
+        # When
+        vaccines = analyzer.getVaccines()
+        
+        # Then
+        self.assertEqual(vaccines, ['6VAX-F', 'ADEN'])
+
+    def test_getSymptoms(self):
+        # Given
+        symptomByVaccine = TestHelper.createDataFrame(
+                columns = ['11-beta-hydroxylase deficiency', '17-hydroxyprogesterone'],
+                data = [  [0.6,                              0.4]],
+                index = pd.Index(
+                    name = 'VAX_TYPE',
+                    data = ['6VAX-F']))
+        
+        analyzer = Analyzer(symptomByVaccine)
+
+        # When
+        symptoms = analyzer.getSymptoms()
+        
+        # Then
+        self.assertEqual(symptoms, ['11-beta-hydroxylase deficiency', '17-hydroxyprogesterone'])