starting CountriesMergerTest

2023-03-31 11:17:21 +02:00
parent 9db4c75765
commit 70e5688be3
2 changed files with 34 additions and 0 deletions
--- a/src/CountriesMergerTest.py
+++ b/src/CountriesMergerTest.py
@@ -0,0 +1,24 @@
+import unittest
+import pandas as pd
+from CountriesMerger import CountriesMerger
+from pandas.testing import assert_series_equal
+
+
+class CountriesMergerTest(unittest.TestCase):
+
+    def test_mergeCountriesSerieses(self):
+        # Given
+        countriesSeriesA = pd.Series({'NO72A': 'CountryA-1', 'EW096': 'CountryA-2'}, name='countriesSeriesA')
+        countriesSeriesB = pd.Series({'NO72A': 'CountryB-1', 'FS192': 'CountryB-2'}, name='countriesSeriesB')
+
+        # When
+        mergedCountries = CountriesMerger.mergeCountriesSerieses(countriesSeriesA, countriesSeriesB)
+
+        # Then
+        assert_series_equal(
+            mergedCountries,
+            pd.Series(
+                {
+                    'NO72A': ['CountryA-1', 'CountryB-1'],
+                    'EW096': ['CountryA-2']
+                }))