starting SymptomHistogramByBatchcodeTableFactoryTest

src/SymptomHistogramByBatchcodeTableFactory.py

@@ -0,0 +1,5 @@
+class SymptomHistogramByBatchcodeTableFactory:
+
+    @staticmethod
+    def createSymptomHistogramByBatchcodeTable(symptomByBatchcodeTable):
+        return symptomByBatchcodeTable.groupby(['VAX_LOT1'])['SYMPTOM'].value_counts().to_frame('SYMPTOM_COUNT_BY_VAX_LOT')

src/SymptomHistogramByBatchcodeTableFactoryTest.py

@@ -0,0 +1,32 @@
+import unittest
+from pandas.testing import assert_frame_equal
+from SymptomHistogramByBatchcodeTableFactory import SymptomHistogramByBatchcodeTableFactory
+from TestHelper import TestHelper
+import pandas as pd
+
+class SymptomHistogramByBatchcodeTableFactoryTest(unittest.TestCase):
+
+    def test_createSymptomHistogramByBatchcodeTable(self):
+        # Given
+        symptomByBatchcodeTable = TestHelper.createDataFrame(
+                columns = ['SYMPTOM'],
+                data = [  ['Blood pressure orthostatic abnormal'],
+                          ['Blood pressure orthostatic abnormal']],
+                index = pd.Index(
+                    name = 'VAX_LOT1',
+                    data = ['EW0175',
+                            'EW0175']))
+                
+        # When
+        symptomHistogramByBatchcodeTable = SymptomHistogramByBatchcodeTableFactory.createSymptomHistogramByBatchcodeTable(symptomByBatchcodeTable)
+
+        # Then
+        assert_frame_equal(
+            symptomHistogramByBatchcodeTable,
+            TestHelper.createDataFrame(
+                columns = ['SYMPTOM_COUNT_BY_VAX_LOT'],
+                data = [  [2]],
+                index = pd.MultiIndex.from_tuples(
+                    names =   ['VAX_LOT1', 'SYMPTOM'],
+                    tuples = [['EW0175',   'Blood pressure orthostatic abnormal']])),
+                check_dtype = False)