Revert "refining CountriesColumnMergerTest"
This reverts commit f9a62a7b44.
This commit is contained in:
@@ -8,8 +8,3 @@ class CountriesColumnMerger:
|
|||||||
dst[COUNTRIES] = CountriesMerger.mergeSrcIntoDst(
|
dst[COUNTRIES] = CountriesMerger.mergeSrcIntoDst(
|
||||||
dst = dst[COUNTRIES],
|
dst = dst[COUNTRIES],
|
||||||
src = src[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')),
|
name = 'VAX_LOT')),
|
||||||
check_dtype = True)
|
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