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

refactoring

Browse Source
This commit is contained in:
frankknoll
2023-03-28 15:01:31 +02:00
parent aaf250b08e
commit f44c0f8869
3 changed files with 92 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/CountriesColumnAdderTest.py
View File

@@ -14,15 +14,16 @@ class CountriesColumnAdderTest(unittest.TestCase):
index = pd.Index(
name = 'Batchcode',
data = ['FE6208']))
columnName = 'Countries guessed'
# When
countriesByBatchcodeTableWithCountriesColumn = CountriesColumnAdder().addCountriesColumn(countriesByBatchcodeTable)
countriesByBatchcodeTableWithCountriesColumn = CountriesColumnAdder().addCountriesColumn(countriesByBatchcodeTable, columnName)
# Then
assert_frame_equal(
countriesByBatchcodeTableWithCountriesColumn,
TestHelper.createDataFrame(
columns = ['United States', 'Germany', 'Italy', 'Countries'],
columns = ['United States', 'Germany', 'Italy', columnName],
data = [ [20, 0, 3, {'United States', 'Italy'}]],
index = pd.Index(
name = 'Batchcode',