starting VaersDescrReaderTest

This commit is contained in:
frankknoll
2023-01-18 17:10:10 +01:00
parent a873d3d692
commit fc62639a1b
4 changed files with 33 additions and 0 deletions

View File

@@ -0,0 +1,25 @@
import unittest
from pandas.testing import assert_frame_equal
from TestHelper import TestHelper
from VaersDescrReader import VaersDescrReader
import pandas as pd
import numpy as np
class VaersDescrReaderTest(unittest.TestCase):
def test_readVaersDescrForYear(self):
# Given
vaersDescrReader = VaersDescrReader(dataDir = "src/testdata")
# When
vaersDescr = vaersDescrReader.readVaersDescrForYear(2023)
# Then
RECVDATE = pd.to_datetime('01/01/2023', format = "%m/%d/%Y")
dataFrameExpected = TestHelper.createDataFrame(
columns = ['RECVDATE', 'DIED', 'L_THREAT', 'ER_VISIT', 'HOSPITAL', 'DISABLE', 'SPLTTYPE'],
data = [ [RECVDATE, np.nan, np.nan, np.nan, np.nan, np.nan, np.nan]],
index = pd.Index(
name = 'VAERS_ID',
data=[2547730]))
assert_frame_equal(vaersDescr['VAERSDATA'], dataFrameExpected, check_dtype=False)