refining SymptomHistogramByBatchcodeTableFactoryTest

This commit is contained in:
frankknoll
2023-01-23 08:51:52 +01:00
parent 4cadd18d45
commit dcdf1f9536

View File

@@ -11,11 +11,13 @@ class SymptomHistogramByBatchcodeTableFactoryTest(unittest.TestCase):
symptomByBatchcodeTable = TestHelper.createDataFrame( symptomByBatchcodeTable = TestHelper.createDataFrame(
columns = ['SYMPTOM'], columns = ['SYMPTOM'],
data = [ ['Blood pressure orthostatic abnormal'], data = [ ['Blood pressure orthostatic abnormal'],
['Blood pressure orthostatic abnormal'],
['Blood pressure orthostatic abnormal']], ['Blood pressure orthostatic abnormal']],
index = pd.Index( index = pd.Index(
name = 'VAX_LOT1', name = 'VAX_LOT1',
data = ['EW0175', data = ['EW0175',
'EW0175'])) 'EW0175',
'1808982']))
# When # When
symptomHistogramByBatchcodeTable = SymptomHistogramByBatchcodeTableFactory.createSymptomHistogramByBatchcodeTable(symptomByBatchcodeTable) symptomHistogramByBatchcodeTable = SymptomHistogramByBatchcodeTableFactory.createSymptomHistogramByBatchcodeTable(symptomByBatchcodeTable)
@@ -25,8 +27,9 @@ class SymptomHistogramByBatchcodeTableFactoryTest(unittest.TestCase):
symptomHistogramByBatchcodeTable, symptomHistogramByBatchcodeTable,
TestHelper.createDataFrame( TestHelper.createDataFrame(
columns = ['SYMPTOM_COUNT_BY_VAX_LOT'], columns = ['SYMPTOM_COUNT_BY_VAX_LOT'],
data = [ [2]], data = [ [1],
[2]],
index = pd.MultiIndex.from_tuples( index = pd.MultiIndex.from_tuples(
names = ['VAX_LOT1', 'SYMPTOM'], names = ['VAX_LOT1', 'SYMPTOM'],
tuples = [['EW0175', 'Blood pressure orthostatic abnormal']])), tuples = [['1808982', 'Blood pressure orthostatic abnormal'],
check_dtype = False) ['EW0175', 'Blood pressure orthostatic abnormal']])))