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

refining CountryColumnsMergerTest

Browse Source
This commit is contained in:
frankknoll
2023-06-13 23:30:24 +02:00
parent 11b2ea4eeb
commit 9e98895553
1 changed files with 15 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
src/CountryColumnsMergerTest.py
View File

@@ -12,10 +12,10 @@ class CountryColumnsMergerTest(unittest.TestCase):
src_val = 'Germany' src_val = 'Germany'
dst_val = None dst_val = None
dst = TestHelper.createDataFrame( dst = TestHelper.createDataFrame(
columns = ['COUNTRY'], columns = ['COUNTRY', 'TestColumn'],
data = [ ['United Kingdom'], data = [ ['United Kingdom', 'test1'],
['France'], ['France', 'test2'],
[dst_val]], [dst_val, 'test3']],
index = pd.Index( index = pd.Index(
name = 'VAERS_ID', name = 'VAERS_ID',
data = [ data = [
@@ -36,31 +36,32 @@ class CountryColumnsMergerTest(unittest.TestCase):
dtypes = {'COUNTRY': 'string'}) dtypes = {'COUNTRY': 'string'})
# When # When
merged = CountryColumnsMerger.mergeCountryColumnOfSrcIntoDst(src = src, dst = dst) merged = CountryColumnsMerger.mergeCountryColumnOfSrcIntoDst(src, dst)
# Then # Then
assert_frame_equal( assert_frame_equal(
merged, merged,
TestHelper.createDataFrame( TestHelper.createDataFrame(
columns = ['COUNTRY'], columns = ['COUNTRY', 'TestColumn'],
data = [ ['United Kingdom'], data = [ ['United Kingdom', 'test1'],
['France'], ['France', 'test2'],
[src_val]], [src_val, 'test3']],
index = pd.Index( index = pd.Index(
name = 'VAERS_ID', name = 'VAERS_ID',
data = [ data = [
'4711', '4711',
'0815', '0815',
'123']), '123']),
dtypes = {'COUNTRY': 'string'})) dtypes = {'COUNTRY': 'string'}),
check_like = True)
def test_shouldNotMergeCountryColumnOfSrcIntoDst_non_unique(self): def test_shouldNotMergeCountryColumnOfSrcIntoDst_non_unique(self):
self._mergeCountryColumnOfSrcIntoDst(val_dst = 'United Kingdom', val_src = 'Germany') self.shouldNotMergeCountryColumnOfSrcIntoDst(val_dst = 'United Kingdom', val_src = 'Germany')
def test_shouldNotMergeCountryColumnOfSrcIntoDst3(self): def test_shouldNotMergeCountryColumnOfSrcIntoDst3(self):
self._mergeCountryColumnOfSrcIntoDst(val_dst = None, val_src = None) self.shouldNotMergeCountryColumnOfSrcIntoDst(val_dst = None, val_src = None)
def _mergeCountryColumnOfSrcIntoDst(self, val_dst, val_src): def shouldNotMergeCountryColumnOfSrcIntoDst(self, val_dst, val_src):
# Given # Given
dst = TestHelper.createDataFrame( dst = TestHelper.createDataFrame(
columns = ['COUNTRY'], columns = ['COUNTRY'],
@@ -82,4 +83,4 @@ class CountryColumnsMergerTest(unittest.TestCase):
# When && Then # When && Then
with self.assertRaises(Exception): with self.assertRaises(Exception):
CountryColumnsMerger.mergeCountryColumnOfSrcIntoDst(src = src, dst = dst) CountryColumnsMerger.mergeCountryColumnOfSrcIntoDst(src, dst)