francois/HowBadIsMyBatch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refining CountryColumnAdderTest

Browse Source
This commit is contained in:
frankknoll
2023-02-07 11:17:47 +01:00
parent 210733c355
commit d7e52bdf49
3 changed files with 82 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

71
src/CountryColumnAdderTest.py
View File

@@ -14,13 +14,16 @@ class CountryColumnAdderTest(unittest.TestCase):
data = [ ['GBPFIZER INC2020486806'],
['FRMODERNATX, INC.MOD20224'],
['dummy']],
index = [
"4711",
"0815",
"123"])
index = pd.Index(
name = 'VAERS_ID',
data = [
"4711",
"0815",
"123"]))
countryColumnAdder = CountryColumnAdder(dataFrame)
# When
dataFrameWithCountryColumn = CountryColumnAdder.addCountryColumn(dataFrame)
dataFrameWithCountryColumn = countryColumnAdder.addCountryColumn(dataFrame)
# Then
assert_frame_equal(
@@ -30,8 +33,56 @@ class CountryColumnAdderTest(unittest.TestCase):
data = [ ['GBPFIZER INC2020486806', 'United Kingdom'],
['FRMODERNATX, INC.MOD20224', 'France'],
['dummy', 'Unknown Country']],
index = [
"4711",
"0815",
"123"],
index = pd.Index(
name = 'VAERS_ID',
data = [
"4711",
"0815",
"123"]),
dtypes = {'COUNTRY': 'string'}))
def test_addCountryColumn2(self):
# Given
countryColumnAdder = CountryColumnAdder(
TestHelper.createDataFrame(
columns = ['SPLTTYPE'],
data = [ ['GBPFIZER INC2020486806'],
['FRMODERNATX, INC.MOD20224'],
['dummy']],
index = pd.Index(
name = 'VAERS_ID',
data = [
2547744,
2547730,
2540815])))
dataFrame = TestHelper.createDataFrame(
columns = ['VAX_LOT'],
data = [ ['1808982'],
['EW0175'],
['EW0176']],
index = pd.Index(
name = 'VAERS_ID',
data = [
2547730,
2547730,
2547744]))
# When
dataFrameWithCountryColumn = countryColumnAdder.addCountryColumn(dataFrame)
# Then
assert_frame_equal(
dataFrameWithCountryColumn,
TestHelper.createDataFrame(
columns = ['VAX_LOT', 'COUNTRY'],
data = [ ['1808982', 'France'],
['EW0175', 'France'],
['EW0176', 'United Kingdom']],
index = pd.Index(
name = 'VAERS_ID',
data = [
2547730,
2547730,
2547744]),
dtypes = {'COUNTRY': 'string'}))