refining VaersDescrReaderTest

2023-01-19 08:04:10 +01:00
parent 28e7eb632a
commit d7193ac1c4
2 changed files with 7 additions and 3 deletions
--- a/src/VaersDescrReaderTest.py
+++ b/src/VaersDescrReaderTest.py
@@ -26,11 +26,14 @@ class VaersDescrReaderTest(unittest.TestCase):
        assert_frame_equal(
                vaersDescr['VAERSVAX'],
                TestHelper.createDataFrame(
-                    columns = ['VAX_TYPE', 'VAX_MANU', 'VAX_LOT', 'VAX_DOSE_SERIES'],
-                    data = [  ['COVID19',  'JANSSEN',  '1808982', 'UNK']],
+                    columns = ['VAX_TYPE', 'VAX_MANU',        'VAX_LOT', 'VAX_DOSE_SERIES'],
+                    data = [  ['COVID19',  'JANSSEN',         '1808982', 'UNK'],
+                              ['COVID19',  'PFIZER\BIONTECH', 'EW0175',  '1']],
                    index = pd.Index(
                            name = 'VAERS_ID',
-                            data=[2547730]),
+                            data=[
+                                2547730,
+                                2547730]),
                    dtypes = {
                        'VAX_DOSE_SERIES': 'string',
                        'VAX_LOT': 'string'}))