refining CountryColumnAdderTest

2023-02-07 11:17:47 +01:00
parent 210733c355
commit d7e52bdf49
3 changed files with 82 additions and 20 deletions
--- a/src/VaersReader.py
+++ b/src/VaersReader.py
@@ -16,7 +16,7 @@ def getVaersForYears(years):
 def getNonDomesticVaers():
    return _getVaers(
        [_getVaersDescrReader().readNonDomesticVaersDescr()],
-        CountryColumnAdder.addCountryColumn)
+        addCountryColumn = lambda dataFrame: CountryColumnAdder(dataFrame).addCountryColumn(dataFrame))

 def _getVaersDescrReader():
    return VaersDescrReader(dataDir = "VAERS")