From d56b1b8a9d5ec56f83674de0083a735d4bea37c5 Mon Sep 17 00:00:00 2001
From: frankknoll <Knoll_Frank@web.de>
Date: Mon, 6 Feb 2023 15:34:51 +0100
Subject: [PATCH] adding CountryColumnAdderTest

---
 src/CountryColumnAdderTest.py | 37 +++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)
 create mode 100644 src/CountryColumnAdderTest.py

diff --git a/src/CountryColumnAdderTest.py b/src/CountryColumnAdderTest.py
new file mode 100644
index 00000000000..38f23955539
--- /dev/null
+++ b/src/CountryColumnAdderTest.py
@@ -0,0 +1,37 @@
+import unittest
+from TestHelper import TestHelper
+from pandas.testing import assert_frame_equal
+import pandas as pd
+from CountryColumnAdder import CountryColumnAdder
+
+
+class CountryColumnAdderTest(unittest.TestCase):
+    
+    def test_addCountryColumn(self):
+        # Given
+        dataFrame = TestHelper.createDataFrame(
+            columns = ['SPLTTYPE'],
+            data = [  ['GBPFIZER INC2020486806'],
+                      ['FRMODERNATX, INC.MOD20224'],
+                      ['dummy']],
+            index = [
+                "4711",
+                "0815",
+                "123"])
+        
+        # When
+        dataFrameWithCountryColumn = CountryColumnAdder.addCountryColumn(dataFrame)
+
+        # Then
+        assert_frame_equal(
+            dataFrameWithCountryColumn,
+            TestHelper.createDataFrame(
+            columns = ['SPLTTYPE',                  'COUNTRY'],
+            data = [  ['GBPFIZER INC2020486806',    'United Kingdom'],
+                      ['FRMODERNATX, INC.MOD20224', 'France'],
+                      ['dummy',                     'Unknown Country']],
+            index = [
+                "4711",
+                "0815",
+                "123"],
+            dtypes = {'COUNTRY': 'string'}))