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

refining CountriesColumnMergerTest

Browse Source
This commit is contained in:
frankknoll
2023-05-30 23:20:49 +02:00
parent bfff2222a3
commit f9a62a7b44
2 changed files with 58 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
src/CountriesColumnMergerTest.py
View File

@@ -51,3 +51,56 @@ class CountriesColumnMergerTest(unittest.TestCase):
],
name = 'VAX_LOT')),
check_dtype = True)
def test_mergeCountriesColumnOfSrcsIntoCountriesColumnOfDst(self):
# Given
dst = TestHelper.createDataFrame(
columns = ['Countries'],
data = [ [[]],
[['France', 'United Kingdom']],
[['France']]],
index = pd.Index(
[
'016M20A',
'030L20B',
'030L20A'
],
name = 'VAX_LOT'))
src1 = TestHelper.createDataFrame(
columns = ['Countries'],
data = [ [['Germany1']]],
index = pd.Index(
[
'030L20B'
],
name = 'VAX_LOT'))
src2 = TestHelper.createDataFrame(
columns = ['Countries'],
data = [ [['Germany2']]],
index = pd.Index(
[
'030L20B'
],
name = 'VAX_LOT'))
# When
CountriesColumnMerger.mergeCountriesColumnOfSrcsIntoCountriesColumnOfDst(dst = dst, srcs = [src1, src2])
# Then
assert_frame_equal(
dst,
TestHelper.createDataFrame(
columns = ['Countries'],
data = [ [[]],
[['France', 'Germany1', 'Germany2', 'United Kingdom']],
[['France']]],
index = pd.Index(
[
'016M20A',
'030L20B',
'030L20A'
],
name = 'VAX_LOT')),
check_dtype = True)