refining CountriesMergerTest

2023-03-31 12:40:05 +02:00
parent 0fa873e461
commit 85c5a0ee60
2 changed files with 3 additions and 3 deletions
--- a/src/CountriesMerger.py
+++ b/src/CountriesMerger.py
@@ -7,7 +7,7 @@ class CountriesMerger:
    def mergeSrcIntoDst(src: pd.Series, dst: pd.Series):
        def merge(series):
            series = series.dropna()
-            return sorted(series[0].union(series[1:]))
+            return sorted(set().union(*series))
        return (pd
                .merge(dst, src, how='left', left_index=True, right_index=True)
                .apply(merge, axis='columns'))