Format UC rebalancing follow-up changes

Author: MaxGhenis

policyengine_uk/scenarios/uc_reform.py

@@ -49,9 +49,7 @@ def _protected_single_young_health_element_monthly(
 ) -> float:
     baseline = sim.tax_benefit_system.parameters(str(BASELINE_UC_REBALANCING_YEAR))
     protected_combined_award = _benefit_uprating_ratio(sim, year) * (
-        float(
-            baseline.gov.dwp.universal_credit.standard_allowance.amount.SINGLE_YOUNG
-        )
+        float(baseline.gov.dwp.universal_credit.standard_allowance.amount.SINGLE_YOUNG)
         + float(baseline.gov.dwp.universal_credit.elements.disabled.amount)
     )
     return protected_combined_award - _rebalanced_standard_allowance_monthly(
@@ -85,9 +83,9 @@ def add_universal_credit_reform(sim: Microsimulation):
             _protected_existing_health_element_monthly(sim, year) * 12,
             dtype=current_health_element.dtype,
         )
-        protected_health_element[
-            claimant_type == UCClaimantType.SINGLE_YOUNG.name
-        ] = _protected_single_young_health_element_monthly(sim, year) * 12
+        protected_health_element[claimant_type == UCClaimantType.SINGLE_YOUNG.name] = (
+            _protected_single_young_health_element_monthly(sim, year) * 12
+        )
         current_health_element[has_health_element & ~is_post_2025_claimant] = (
             protected_health_element[has_health_element & ~is_post_2025_claimant]
         )

policyengine_uk/tests/test_uc_rebalancing.py

@@ -70,22 +70,18 @@ def _cpi_protected_uc_award_monthly(
 
 
 @pytest.mark.parametrize("age_2025", [20, 30])
-def test_existing_claimants_keep_combined_award_cpi_protected(
-    monkeypatch, age_2025
-):
+def test_existing_claimants_keep_combined_award_cpi_protected(monkeypatch, age_2025):
     _force_uc_seed(monkeypatch, [0.99])
     sim = Simulation(situation=_uc_claimant(age_2025))
     claimant_type = "SINGLE_YOUNG" if age_2025 < 25 else "SINGLE_OLD"
 
     for year in range(2026, 2030):
         standard_allowance = sim.calculate("uc_standard_allowance", year)[0] / 12
         health_element = sim.calculate("uc_LCWRA_element", year)[0] / 12
-        new_claimant_health_element = (
-            float(
-                sim.tax_benefit_system.parameters(
-                    str(year)
-                ).gov.dwp.universal_credit.rebalancing.new_claimant_health_element
-            )
+        new_claimant_health_element = float(
+            sim.tax_benefit_system.parameters(
+                str(year)
+            ).gov.dwp.universal_credit.rebalancing.new_claimant_health_element
         )
 
         assert health_element > new_claimant_health_element
@@ -100,12 +96,10 @@ def test_new_claimants_use_fixed_health_element(monkeypatch):
 
     for year in range(2026, 2030):
         health_element = sim.calculate("uc_LCWRA_element", year)[0] / 12
-        expected_health = (
-            float(
-                sim.tax_benefit_system.parameters(
-                    str(year)
-                ).gov.dwp.universal_credit.rebalancing.new_claimant_health_element
-            )
+        expected_health = float(
+            sim.tax_benefit_system.parameters(
+                str(year)
+            ).gov.dwp.universal_credit.rebalancing.new_claimant_health_element
         )
 
         assert health_element == pytest.approx(expected_health)