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

adding CountriesColumnMerger

Browse Source
This commit is contained in:
frankknoll
2023-05-30 23:00:06 +02:00
parent 7c529bb9b7
commit 2b45597d08
2 changed files with 63 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/CountriesColumnMerger.py Normal file
View File

@@ -0,0 +1,10 @@
from CountriesMerger import CountriesMerger
class CountriesColumnMerger:
@staticmethod
def mergeCountriesColumnOfSrcIntoCountriesColumnOfDst(src, dst):
COUNTRIES = 'Countries'
dst[COUNTRIES] = CountriesMerger.mergeSrcIntoDst(
dst = dst[COUNTRIES],
src = src[COUNTRIES])

53
src/CountriesColumnMergerTest.py Normal file
View File

@@ -0,0 +1,53 @@
import unittest
import pandas as pd
from pandas.testing import assert_frame_equal
from TestHelper import TestHelper
from src.CountriesColumnMerger import CountriesColumnMerger
class CountriesColumnMergerTest(unittest.TestCase):
def test_mergeCountriesColumnOfSrcIntoCountriesColumnOfDst(self):
# Given
dst = TestHelper.createDataFrame(
columns = ['Countries'],
data = [ [[]],
[['France', 'United Kingdom']],
[['France']]],
index = pd.Index(
[
'016M20A',
'030L20B',
'030L20A'
],
name = 'VAX_LOT'))
src = TestHelper.createDataFrame(
columns = ['Countries'],
data = [ [['Germany']],
[['DummyCountry']]],
index = pd.Index(
[
'030L20B',
'dummyVaxLot'
],
name = 'VAX_LOT'))
# When
CountriesColumnMerger.mergeCountriesColumnOfSrcIntoCountriesColumnOfDst(dst = dst, src = src)
# Then
assert_frame_equal(
dst,
TestHelper.createDataFrame(
columns = ['Countries'],
data = [ [[]],
[['France', 'Germany', 'United Kingdom']],
[['France']]],
index = pd.Index(
[
'016M20A',
'030L20B',
'030L20A'
],
name = 'VAX_LOT')),
check_dtype = True)