refining CountriesMergerTest

2023-04-01 01:15:00 +02:00
parent 49db8cde8f
commit 0b231d12dd
2 changed files with 8 additions and 4 deletions
--- a/src/CountriesMerger.py
+++ b/src/CountriesMerger.py
@@ -8,6 +8,9 @@ class CountriesMerger:
        def merge(series):
            series = series.dropna()
            return sorted(set().union(*series))
-        return (pd
-                .merge(dst, src, how='left', left_index=True, right_index=True)
-                .apply(merge, axis='columns'))
+        
+        mergedSeries = (pd
+                        .merge(dst, src, how='left', left_index=True, right_index=True)
+                        .apply(merge, axis='columns'))
+        mergedSeries.name = dst.name
+        return mergedSeries