30 lines
1.1 KiB
Python
30 lines
1.1 KiB
Python
import unittest
|
|
from TestHelper import TestHelper
|
|
from pandas.testing import assert_frame_equal
|
|
import pandas as pd
|
|
from src.CountriesColumnAdder import CountriesColumnAdder
|
|
|
|
class CountriesColumnAdderTest(unittest.TestCase):
|
|
|
|
def test_addCountriesColumn(self):
|
|
# Given
|
|
countriesByBatchcodeTable = TestHelper.createDataFrame(
|
|
columns = ['United States', 'Germany', 'Italy'],
|
|
data = [ [20, 0, 3]],
|
|
index = pd.Index(
|
|
name = 'Batchcode',
|
|
data = ['FE6208']))
|
|
|
|
# When
|
|
countriesByBatchcodeTableWithCountriesColumn = CountriesColumnAdder().addCountriesColumn(countriesByBatchcodeTable)
|
|
|
|
# Then
|
|
assert_frame_equal(
|
|
countriesByBatchcodeTableWithCountriesColumn,
|
|
TestHelper.createDataFrame(
|
|
columns = ['United States', 'Germany', 'Italy', 'Countries'],
|
|
data = [ [20, 0, 3, {'United States', 'Italy'}]],
|
|
index = pd.Index(
|
|
name = 'Batchcode',
|
|
data = ['FE6208'])))
|
|
|