refining AnalyzerTest

2023-10-09 11:36:03 +02:00
parent e1d26752fd
commit f658bafb8a
4 changed files with 40 additions and 1 deletions
--- a/src/SymptomsCausedByVaccines/Analyzer.py
+++ b/src/SymptomsCausedByVaccines/Analyzer.py
@@ -7,3 +7,6 @@ class Analyzer:

    def getSymptomsForVaccine(self, vaxType):
        return self.symptomByVaccine.loc[vaxType]
+
+    def getVaccinesForSymptom(self, symptom):
+        return self.symptomByVaccine[symptom]