starting SymptomHistogramByBatchcodeTableFactoryTest

2023-01-23 08:34:26 +01:00
parent b6eb929a7b
commit 4cadd18d45
2 changed files with 37 additions and 0 deletions
--- a/src/SymptomHistogramByBatchcodeTableFactoryTest.py
+++ b/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)