refactoring

This commit is contained in:
frankknoll
2023-06-02 13:50:18 +02:00
parent 09905eed4d
commit b0142cef77

View File

@@ -24,11 +24,15 @@ class CountryCountsByBatchcodeTablesMergerTest(unittest.TestCase):
tuples = [['12345', 'Germany']]))
# When
dataFrame = CountryCountsByBatchcodeTablesMerger.merge([countryCountsByBatchcodeTable1, countryCountsByBatchcodeTable2])
mergedCountryCountsByBatchcodeTables = CountryCountsByBatchcodeTablesMerger.merge(
[
countryCountsByBatchcodeTable1,
countryCountsByBatchcodeTable2
])
# Then
assert_frame_equal(
dataFrame,
mergedCountryCountsByBatchcodeTables,
TestHelper.createDataFrame(
columns = ['COUNTRY_COUNT_BY_VAX_LOT'],
data = [ [10 + 20],