From 0b231d12dd5bfdb16c1c59ec60de167c45769a6c Mon Sep 17 00:00:00 2001 From: frankknoll Date: Sat, 1 Apr 2023 01:15:00 +0200 Subject: [PATCH] refining CountriesMergerTest --- src/CountriesMerger.py | 9 ++++++--- src/CountriesMergerTest.py | 3 ++- 2 files changed, 8 insertions(+), 4 deletions(-) diff --git a/src/CountriesMerger.py b/src/CountriesMerger.py index dd17eed83d0..3f703ced405 100644 --- 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 diff --git a/src/CountriesMergerTest.py b/src/CountriesMergerTest.py index a5a7a2272e2..dd12ff99406 100644 --- a/src/CountriesMergerTest.py +++ b/src/CountriesMergerTest.py @@ -21,4 +21,5 @@ class CountriesMergerTest(unittest.TestCase): { 'NO72A': ['CountryA-1', 'CountryB-1'], 'EW096': ['CountryA-2'] - })) + }, + name=dstCountries.name))