Revert "refining CountriesColumnMergerTest"
This reverts commit f9a62a7b44.
This commit is contained in:
@@ -8,8 +8,3 @@ 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)
|
||||
|
||||
@@ -51,56 +51,3 @@ 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)
|
||||
|
||||
Reference in New Issue
Block a user