refining BatchcodeCompletionTest

2023-03-24 17:57:31 +01:00
parent 0d4006ea92
commit 918215cd40
1 changed files with 16 additions and 0 deletions
--- a/src/BatchcodeCompletionTest.py
+++ b/src/BatchcodeCompletionTest.py
@@ -48,3 +48,19 @@ class BatchcodeCompletionTest(unittest.TestCase):

        # Then
        self.assertIsNone(completedBatchcode)
+
+    def test_completeBatchcode_empty_ADR_by_Batchcode(self):
+        # Given
+        ADR_by_Batchcode = TestHelper.createDataFrame(
+                columns = ['Adverse Reaction Reports'],
+                data = [  ],
+                index = pd.Index(
+                    [],
+                    name = 'VAX_LOT'))
+        batchcodeCompletion = BatchcodeCompletion(ADR_by_Batchcode)
+                        
+        # When
+        completedBatchcode = batchcodeCompletion.completeBatchcode('non existing batch code')
+
+        # Then
+        self.assertIsNone(completedBatchcode)