-
Notifications
You must be signed in to change notification settings - Fork 6
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
9412b9d
commit 06f0d2c
Showing
4 changed files
with
289 additions
and
23 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
77 changes: 77 additions & 0 deletions
77
policyengine/economic_impact/labour_supply_impact/earnings/overall/absolute/absolute.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,77 @@ | ||
| from policyengine.economic_impact.base_metric_calculator import BaseMetricCalculator | ||
| from policyengine_uk import Microsimulation | ||
| from policyengine_core.reforms import Reform | ||
| from microdf import MicroSeries | ||
| import numpy as np | ||
|
|
||
|
|
||
| class IncomeLSR(BaseMetricCalculator): | ||
| def __init__(self, baseline: Microsimulation, reformed: Microsimulation, default_period: int = 2024) -> None: | ||
| super().__init__(baseline, reformed, default_period) | ||
| self.baseline = baseline | ||
| self.reformed = reformed | ||
|
|
||
| def calculate(self): | ||
| income_lsr_baseline = 0 | ||
| income_lsr_reformed = 0 | ||
|
|
||
| if "employment_income_behavioral_response" in self.baseline.tax_benefit_system.variables: | ||
| if any(self.baseline.calculate("employment_income_behavioral_response") != 0): | ||
| income_lsr_baseline = self.baseline.calculate("income_elasticity_lsr").sum() | ||
|
|
||
| if "employment_income_behavioral_response" in self.reformed.tax_benefit_system.variables: | ||
| if any(self.reformed.calculate("employment_income_behavioral_response") != 0): | ||
| income_lsr_reformed = self.reformed.calculate("income_elasticity_lsr").sum() | ||
|
|
||
| income_lsr = income_lsr_reformed - income_lsr_baseline | ||
| # Convert to billions and round to 2 decimal places | ||
| income_lsr_billion = round(income_lsr / 1e9, 2) | ||
|
|
||
| return { | ||
| "income_lsr": income_lsr_billion | ||
| } | ||
|
|
||
| class SubstitutionLSR(BaseMetricCalculator): | ||
| def __init__(self, baseline: Microsimulation, reformed: Microsimulation, default_period: int = 2024) -> None: | ||
| super().__init__(baseline, reformed, default_period) | ||
| self.baseline = baseline | ||
| self.reformed = reformed | ||
|
|
||
| def calculate(self): | ||
| substitution_lsr_baseline = 0 | ||
| substitution_lsr_reformed = 0 | ||
|
|
||
| if "employment_income_behavioral_response" in self.baseline.tax_benefit_system.variables: | ||
| if any(self.baseline.calculate("employment_income_behavioral_response") != 0): | ||
| substitution_lsr_baseline = self.baseline.calculate("substitution_elasticity_lsr").sum() | ||
|
|
||
| if "employment_income_behavioral_response" in self.reformed.tax_benefit_system.variables: | ||
| if any(self.reformed.calculate("employment_income_behavioral_response") != 0): | ||
| substitution_lsr_reformed = self.reformed.calculate("substitution_elasticity_lsr").sum() | ||
|
|
||
| substitution_lsr = substitution_lsr_reformed - substitution_lsr_baseline | ||
| # Convert to billions and round to 2 decimal places | ||
| substitution_lsr_billion = round(substitution_lsr / 1e9, 2) | ||
|
|
||
| return { | ||
| "substitution_lsr": substitution_lsr_billion | ||
| } | ||
|
|
||
| class NetLSRChange(BaseMetricCalculator): | ||
| def __init__(self, baseline: Microsimulation, reformed: Microsimulation, default_period: int = 2024) -> None: | ||
| super().__init__(baseline, reformed, default_period) | ||
| self.baseline = baseline | ||
| self.reformed = reformed | ||
|
|
||
| def calculate(self): | ||
| income_result = IncomeLSR(baseline=self.baseline, reformed=self.reformed).calculate() | ||
| substitution_result = SubstitutionLSR(baseline=self.baseline, reformed=self.reformed).calculate() | ||
|
|
||
| income_value = income_result["income_lsr"] | ||
| substitution_value = substitution_result["substitution_lsr"] | ||
|
|
||
| net_change = (income_value + substitution_value) | ||
| # Round to 2 decimal places | ||
| net_change_billion = round(net_change, 2) | ||
|
|
||
| return {"net_change": net_change_billion} |
143 changes: 143 additions & 0 deletions
143
policyengine/economic_impact/labour_supply_impact/earnings/overall/relative/relative.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,143 @@ | ||
| from policyengine.economic_impact.base_metric_calculator import BaseMetricCalculator | ||
| from policyengine_uk import Microsimulation | ||
| from policyengine_core.reforms import Reform | ||
| from microdf import MicroSeries | ||
| import numpy as np | ||
|
|
||
| class IncomeLSR(BaseMetricCalculator): | ||
| def __init__(self, baseline: Microsimulation, reformed: Microsimulation, default_period: int = 2024) -> None: | ||
| super().__init__(baseline, reformed, default_period) | ||
| self.baseline = baseline | ||
| self.reformed = reformed | ||
|
|
||
| def calculate(self): | ||
| # Calculate household counts | ||
| # household_count_people_baseline = self.baseline.calculate("household_count_people") | ||
| # household_count_people_reformed = self.reformed.calculate("household_count_people") | ||
|
|
||
| # Calculate baseline and reformed substitution LSR | ||
| substitution_lsr_hh_baseline = self._calculate_substitution_lsr(self.baseline) | ||
| substitution_lsr_hh_reformed = self._calculate_substitution_lsr(self.reformed) | ||
|
|
||
| # Compute the change in substitution LSR | ||
| substitution_lsr_hh = np.array(substitution_lsr_hh_reformed) - np.array(substitution_lsr_hh_baseline) | ||
|
|
||
| # Calculate baseline and reformed income LSR | ||
| income_lsr_hh_baseline = self._calculate_income_lsr(self.baseline) | ||
| income_lsr_hh_reformed = self._calculate_income_lsr(self.reformed) | ||
|
|
||
| # Compute the change in income LSR | ||
| income_lsr_hh = np.array(income_lsr_hh_reformed) - np.array(income_lsr_hh_baseline) | ||
|
|
||
| # Calculate weights and earnings | ||
| household_weight = self.baseline.calculate("household_weight") | ||
| emp_income = MicroSeries( | ||
| self.baseline.calculate("employment_income", map_to="household").astype(float).tolist(), | ||
| weights=household_weight | ||
| ) | ||
| self_emp_income = MicroSeries( | ||
| self.baseline.calculate("self_employment_income", map_to="household").astype(float).tolist(), | ||
| weights=household_weight | ||
| ) | ||
| total_lsr_hh = substitution_lsr_hh + income_lsr_hh | ||
| earnings = emp_income + self_emp_income | ||
| original_earnings = earnings - total_lsr_hh | ||
|
|
||
| # Calculate income LSR ratio | ||
| income_lsr_hh = MicroSeries(income_lsr_hh, weights=household_weight) | ||
| income = income_lsr_hh.sum() / original_earnings.sum() * 100 | ||
|
|
||
| return {"income": round(income,2)} | ||
|
|
||
| def _calculate_substitution_lsr(self, simulation: Microsimulation): | ||
| """Calculate substitution LSR for a given simulation.""" | ||
| if "employment_income_behavioral_response" in simulation.tax_benefit_system.variables: | ||
| if any(simulation.calculate("employment_income_behavioral_response") != 0): | ||
| return simulation.calculate("substitution_elasticity_lsr", map_to="household").astype(float).tolist() | ||
| return (self.baseline.calculate("household_count_people") * 0).astype(float).tolist() | ||
|
|
||
| def _calculate_income_lsr(self, simulation: Microsimulation): | ||
| """Calculate income LSR for a given simulation.""" | ||
| if "employment_income_behavioral_response" in simulation.tax_benefit_system.variables: | ||
| if any(simulation.calculate("employment_income_behavioral_response") != 0): | ||
| return simulation.calculate("income_elasticity_lsr", map_to="household").astype(float).tolist() | ||
| return (self.baseline.calculate("household_count_people") * 0).astype(float).tolist() | ||
|
|
||
|
|
||
| class SubstitutionLSR(BaseMetricCalculator): | ||
| def __init__(self, baseline: Microsimulation, reformed: Microsimulation, default_period: int = 2024) -> None: | ||
| super().__init__(baseline, reformed, default_period) | ||
| self.baseline = baseline | ||
| self.reformed = reformed | ||
|
|
||
| def calculate(self): | ||
| # Calculate household counts | ||
| # household_count_people_baseline = self.baseline.calculate("household_count_people") | ||
| # household_count_people_reformed = self.reformed.calculate("household_count_people") | ||
|
|
||
| # Calculate baseline and reformed income LSR | ||
| income_lsr_hh_baseline = self._calculate_income_lsr(self.baseline) | ||
| income_lsr_hh_reformed = self._calculate_income_lsr(self.reformed) | ||
|
|
||
| # Calculate baseline and reformed substitution LSR | ||
| substitution_lsr_hh_baseline = self._calculate_substitution_lsr(self.baseline) | ||
| substitution_lsr_hh_reformed = self._calculate_substitution_lsr(self.reformed) | ||
|
|
||
| # Compute the change in substitution LSR | ||
| substitution_lsr_hh = np.array(substitution_lsr_hh_reformed) - np.array(substitution_lsr_hh_baseline) | ||
| household_weight = self.baseline.calculate("household_weight") | ||
|
|
||
| # Convert to MicroSeries and compute total LSR | ||
| substitution_lsr_hh = MicroSeries(substitution_lsr_hh, weights=household_weight) | ||
| income_lsr_hh = np.array(income_lsr_hh_reformed) - np.array(income_lsr_hh_baseline) | ||
| total_lsr_hh = substitution_lsr_hh + income_lsr_hh | ||
|
|
||
| # Calculate earnings | ||
| emp_income = MicroSeries( | ||
| self.baseline.calculate("employment_income", map_to="household").astype(float).tolist(), | ||
| weights=household_weight | ||
| ) | ||
| self_emp_income = MicroSeries( | ||
| self.baseline.calculate("self_employment_income", map_to="household").astype(float).tolist(), | ||
| weights=household_weight | ||
| ) | ||
| earnings = emp_income + self_emp_income | ||
| original_earnings = earnings - total_lsr_hh | ||
|
|
||
| # Calculate substitution ratio | ||
| substitution = substitution_lsr_hh.sum() / original_earnings.sum() * 100 | ||
|
|
||
| return {"substitution": round(substitution,2)} | ||
|
|
||
| def _calculate_substitution_lsr(self, simulation: Microsimulation): | ||
| """Calculate substitution LSR for a given simulation.""" | ||
| if "employment_income_behavioral_response" in simulation.tax_benefit_system.variables: | ||
| if any(simulation.calculate("employment_income_behavioral_response") != 0): | ||
| return simulation.calculate("substitution_elasticity_lsr", map_to="household").astype(float).tolist() | ||
| return (self.baseline.calculate("household_count_people") * 0).astype(float).tolist() | ||
|
|
||
| def _calculate_income_lsr(self, simulation: Microsimulation): | ||
| """Calculate income LSR for a given simulation.""" | ||
| if "employment_income_behavioral_response" in simulation.tax_benefit_system.variables: | ||
| if any(simulation.calculate("employment_income_behavioral_response") != 0): | ||
| return simulation.calculate("income_elasticity_lsr", map_to="household").astype(float).tolist() | ||
| return (self.baseline.calculate("household_count_people") * 0).astype(float).tolist() | ||
|
|
||
|
|
||
|
|
||
| class NetLSRChange(BaseMetricCalculator): | ||
| def __init__(self, baseline: Microsimulation, reformed: Microsimulation, default_period: int = 2024) -> None: | ||
| super().__init__(baseline, reformed, default_period) | ||
| self.baseline = baseline | ||
| self.reformed = reformed | ||
|
|
||
| def calculate(self): | ||
| income_result = IncomeLSR(baseline=self.baseline, reformed=self.reformed).calculate() | ||
| substitution_result = SubstitutionLSR(baseline=self.baseline, reformed=self.reformed).calculate() | ||
|
|
||
| income_value = income_result["income"] | ||
| substitution_value = substitution_result["substitution"] | ||
|
|
||
| net_change = (income_value + substitution_value) | ||
|
|
||
| return {"net_change": round(net_change,2)} |