Adding ETR computations to france-data

openfisca_france_data/erfs_fpr/get_survey_scenario.py

@@ -19,6 +19,9 @@ def get_survey_scenario(
         baseline_tax_benefit_system: Optional[TaxBenefitSystem] = None,
         data: Any = None,  # Plusiers formats possibles.
         reform: Optional[Reform] = None,
+        use_marginal_tax_rate: bool = False,
+        variation_factor: float = 0.03,
+        varying_variable: str = None,
         ) -> ErfsFprSurveyScenario:
     """Helper pour créer un :class:`ErfsFprSurveyScenario`.
 
@@ -38,11 +41,21 @@ def get_survey_scenario(
         baseline_tax_benefit_system,
         )
 
-    survey_scenario = ErfsFprSurveyScenario.create(
-        tax_benefit_system = tax_benefit_system,
-        baseline_tax_benefit_system = baseline_tax_benefit_system,
-        year = year,
-        )
+    if use_marginal_tax_rate == False:
+        survey_scenario = ErfsFprSurveyScenario.create(
+            tax_benefit_system = tax_benefit_system,
+            baseline_tax_benefit_system = baseline_tax_benefit_system,
+            year = year,
+            )
+    else:
+        assert varying_variable is not None, "You need to specify the varying variable."
+        survey_scenario = ErfsFprSurveyScenario.create(
+            tax_benefit_system = tax_benefit_system,
+            baseline_tax_benefit_system = baseline_tax_benefit_system,
+            year = year,
+            )
+        survey_scenario.variation_factor = variation_factor
+        survey_scenario.varying_variable = varying_variable
 
     # S'il n'y a pas de données, on sait où les trouver.
     if data is None:
@@ -70,6 +83,7 @@ def get_survey_scenario(
     survey_scenario.init_from_data(
         data = data,
         rebuild_input_data = rebuild_input_data,
+        use_marginal_tax_rate = use_marginal_tax_rate,
         )
 
     return survey_scenario