refactoring

2023-01-27 09:17:52 +01:00
parent cb2dffcb57
commit d56b17fd17
2 changed files with 11 additions and 13 deletions
--- a/src/DictByBatchcodeTable2DictConverterTest.py
+++ b/src/DictByBatchcodeTable2DictConverterTest.py
@@ -0,0 +1,53 @@
+import unittest
+import json
+from DictByBatchcodeTable2DictConverter import DictByBatchcodeTable2DictConverter
+from TestHelper import TestHelper
+import pandas as pd
+
+class DictByBatchcodeTable2DictConverterTest(unittest.TestCase):
+
+    def test_convertDictByBatchcodeTable2Json(self):
+        # Given
+        dictByBatchcodeTable = TestHelper.createDataFrame(
+                columns = ['SYMPTOM_COUNT_BY_VAX_LOT'],
+                data = [  [
+                            {
+                                "Blood pressure orthostatic abnormal": 5,
+                                "Chest discomfort": 1
+                            }
+                          ],
+                          [
+                            {
+                                "Chest discomfort": 2
+                            }
+                          ]
+                        ],
+                index = pd.MultiIndex.from_tuples(
+                    names =   ['VAX_LOT1', 'VAX_LOT2', 'VAX_LOT3'],
+                    tuples = [['1808982',  'EW0175',   'FD1921'],
+                              ['015M20A',  '1808982',  'nan']]))
+
+        # When
+        dict = DictByBatchcodeTable2DictConverter.convertDictByBatchcodeTable2Dict(dictByBatchcodeTable, '1808982')
+
+        # Then
+        self.assertEqual(
+            dict,
+                {
+                    "batchcode": "1808982",
+                    "histograms": [
+                        {
+                            "batchcodes": ["1808982", "EW0175", "FD1921"],
+                            "histogram": {
+                                "Blood pressure orthostatic abnormal": 5,
+                                "Chest discomfort": 1
+                            }
+                        },
+                        {
+                            "batchcodes": ["015M20A", "1808982"],
+                            "histogram": {
+                                "Chest discomfort": 2
+                            }
+                        }
+                    ]
+                })