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

replacing 'Unknown Country' by None

Browse Source
This commit is contained in:
frankknoll
2023-04-01 11:39:36 +02:00
parent 74ebd55a47
commit e7256697c3
5 changed files with 27 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
src/CountryColumnAdderTest.py
View File

@@ -29,17 +29,17 @@ class CountryColumnAdderTest(unittest.TestCase):
assert_frame_equal(
dataFrameWithCountryColumn,
TestHelper.createDataFrame(
columns = ['SPLTTYPE', 'COUNTRY'],
data = [ ['GBPFIZER INC2020486806', 'United Kingdom'],
['FRMODERNATX, INC.MOD20224', 'France'],
['dummy', 'Unknown Country']],
index = pd.Index(
name = 'VAERS_ID',
data = [
"4711",
"0815",
"123"]),
dtypes = {'COUNTRY': 'string'}))
columns = ['SPLTTYPE', 'COUNTRY'],
data = [ ['GBPFIZER INC2020486806', 'United Kingdom'],
['FRMODERNATX, INC.MOD20224', 'France'],
['dummy', None]],
index = pd.Index(
name = 'VAERS_ID',
data = [
"4711",
"0815",
"123"]),
dtypes = {'COUNTRY': 'string'}))
def test_addCountryColumn2(self):
@@ -75,14 +75,14 @@ class CountryColumnAdderTest(unittest.TestCase):
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'}))
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'}))