From dcdf1f9536d362199e5e141ca0755e0a3152a657 Mon Sep 17 00:00:00 2001 From: frankknoll Date: Mon, 23 Jan 2023 08:51:52 +0100 Subject: [PATCH] refining SymptomHistogramByBatchcodeTableFactoryTest --- src/SymptomHistogramByBatchcodeTableFactoryTest.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/src/SymptomHistogramByBatchcodeTableFactoryTest.py b/src/SymptomHistogramByBatchcodeTableFactoryTest.py index 9ab3f3bac43..37429b852a6 100644 --- a/src/SymptomHistogramByBatchcodeTableFactoryTest.py +++ b/src/SymptomHistogramByBatchcodeTableFactoryTest.py @@ -11,11 +11,13 @@ class SymptomHistogramByBatchcodeTableFactoryTest(unittest.TestCase): symptomByBatchcodeTable = TestHelper.createDataFrame( columns = ['SYMPTOM'], data = [ ['Blood pressure orthostatic abnormal'], + ['Blood pressure orthostatic abnormal'], ['Blood pressure orthostatic abnormal']], index = pd.Index( name = 'VAX_LOT1', data = ['EW0175', - 'EW0175'])) + 'EW0175', + '1808982'])) # When symptomHistogramByBatchcodeTable = SymptomHistogramByBatchcodeTableFactory.createSymptomHistogramByBatchcodeTable(symptomByBatchcodeTable) @@ -25,8 +27,9 @@ class SymptomHistogramByBatchcodeTableFactoryTest(unittest.TestCase): symptomHistogramByBatchcodeTable, TestHelper.createDataFrame( columns = ['SYMPTOM_COUNT_BY_VAX_LOT'], - data = [ [2]], + data = [ [1], + [2]], index = pd.MultiIndex.from_tuples( names = ['VAX_LOT1', 'SYMPTOM'], - tuples = [['EW0175', 'Blood pressure orthostatic abnormal']])), - check_dtype = False) + tuples = [['1808982', 'Blood pressure orthostatic abnormal'], + ['EW0175', 'Blood pressure orthostatic abnormal']])))