refactoring
This commit is contained in:
@@ -24,11 +24,15 @@ class CountryCountsByBatchcodeTablesMergerTest(unittest.TestCase):
|
|||||||
tuples = [['12345', 'Germany']]))
|
tuples = [['12345', 'Germany']]))
|
||||||
|
|
||||||
# When
|
# When
|
||||||
dataFrame = CountryCountsByBatchcodeTablesMerger.merge([countryCountsByBatchcodeTable1, countryCountsByBatchcodeTable2])
|
mergedCountryCountsByBatchcodeTables = CountryCountsByBatchcodeTablesMerger.merge(
|
||||||
|
[
|
||||||
|
countryCountsByBatchcodeTable1,
|
||||||
|
countryCountsByBatchcodeTable2
|
||||||
|
])
|
||||||
|
|
||||||
# Then
|
# Then
|
||||||
assert_frame_equal(
|
assert_frame_equal(
|
||||||
dataFrame,
|
mergedCountryCountsByBatchcodeTables,
|
||||||
TestHelper.createDataFrame(
|
TestHelper.createDataFrame(
|
||||||
columns = ['COUNTRY_COUNT_BY_VAX_LOT'],
|
columns = ['COUNTRY_COUNT_BY_VAX_LOT'],
|
||||||
data = [ [10 + 20],
|
data = [ [10 + 20],
|
||||||
|
|||||||
Reference in New Issue
Block a user