refactoring

This commit is contained in:
frankknoll
2023-03-24 23:39:26 +01:00
parent 0ec8d406c9
commit 7f0d5b66fb

View File

@@ -52,18 +52,11 @@ class BatchcodeCompletionTest(unittest.TestCase):
def test_completeBatchcode_NaN(self): def test_completeBatchcode_NaN(self):
# Given # Given
# FK-TODO: use empty ADR_by_Batchcode
ADR_by_Batchcode = TestHelper.createDataFrame( ADR_by_Batchcode = TestHelper.createDataFrame(
columns = ['Adverse Reaction Reports'], columns = ['Adverse Reaction Reports'],
data = [ [1], data = [],
[200],
[149]],
index = pd.Index( index = pd.Index(
[ [],
'LOT000057A',
'030L20B',
'000057A'
],
name = 'VAX_LOT')) name = 'VAX_LOT'))
batchcodeCompletion = BatchcodeCompletion(ADR_by_Batchcode) batchcodeCompletion = BatchcodeCompletion(ADR_by_Batchcode)
@@ -77,7 +70,7 @@ class BatchcodeCompletionTest(unittest.TestCase):
# Given # Given
ADR_by_Batchcode = TestHelper.createDataFrame( ADR_by_Batchcode = TestHelper.createDataFrame(
columns = ['Adverse Reaction Reports'], columns = ['Adverse Reaction Reports'],
data = [ ], data = [],
index = pd.Index( index = pd.Index(
[], [],
name = 'VAX_LOT')) name = 'VAX_LOT'))