adding VaccineDistributionByZipcodeSimplifier

2023-03-09 20:38:17 +01:00
parent b089ca98f3
commit 483e64ef43
5 changed files with 62 additions and 3 deletions
--- a/src/TestHelper.py
+++ b/src/TestHelper.py
@@ -4,5 +4,5 @@ import pandas as pd
 class TestHelper:

    @staticmethod
-    def createDataFrame(index, columns, data, dtypes={}):
-        return pd.DataFrame(index=index, columns=columns, data=data).astype(dtypes)
+    def createDataFrame(columns, data, dtypes={}, **kwargs):
+        return pd.DataFrame(columns=columns, data=data, **kwargs).astype(dtypes)