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

Revert "Revert "refining CountriesColumnMergerTest""

Browse Source 
This reverts commit 531475a7f5.
This commit is contained in:
frankknoll
2023-05-31 19:32:45 +02:00
parent a213e0a059
commit 360d8094ad
2 changed files with 58 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/CountriesColumnMerger.py
View File

@@ -8,3 +8,8 @@ class CountriesColumnMerger:
dst[COUNTRIES] = CountriesMerger.mergeSrcIntoDst(
dst = dst[COUNTRIES],
src = src[COUNTRIES])
@staticmethod
def mergeCountriesColumnOfSrcsIntoCountriesColumnOfDst(srcs, dst):
for src in srcs:
CountriesColumnMerger.mergeCountriesColumnOfSrcIntoCountriesColumnOfDst(src = src, dst = dst)

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)