formatted and change logged

Author: SakshiKekre

changelog_entry.yaml

@@ -0,0 +1,5 @@
+- bump: patch
+  changes:
+    added:
+    - Unit test for GeneralEconomyTask.calculate_cliffs() and fixture for the test
+    - Test for calculate_single_economy with cliff impacts

tests/country/test_us.py

@@ -22,6 +22,7 @@ def test_us_macro_comparison():
 
     sim.calculate_economy_comparison()
 
+
 def test_us_macro_cliff_impacts():
     from policyengine import Simulation
 
@@ -31,13 +32,15 @@ def test_us_macro_cliff_impacts():
         reform={
             "gov.usda.snap.income.deductions.earned_income": {"2025": 0.05}
         },
-        include_cliffs=True
+        include_cliffs=True,
     )
 
     result = sim.calculate_economy_comparison()
-    cliff_impact = result.model_dump().get('cliff_impact')
+    cliff_impact = result.model_dump().get("cliff_impact")
 
-    assert cliff_impact is not None, "Expected 'cliff_impact' to be present in the output."
+    assert (
+        cliff_impact is not None
+    ), "Expected 'cliff_impact' to be present in the output."
 
     assert cliff_impact["baseline"]["cliff_gap"] > 0
     assert cliff_impact["reform"]["cliff_share"] > 0

tests/test_simulation.py

@@ -11,7 +11,9 @@
 from copy import deepcopy
 
 from policyengine import Simulation
-from policyengine.outputs.macro.single.calculate_single_economy import GeneralEconomyTask
+from policyengine.outputs.macro.single.calculate_single_economy import (
+    GeneralEconomyTask,
+)
 
 
 class TestSimulation:
@@ -73,9 +75,10 @@ def test__given_dataset_without_time_period__does_not_set_time_period(
                 == None
             )
 
-
     class TestCalculateCliffs:
-        def test__calculates_correct_cliff_metrics(self, mock_simulation_with_cliff_vars):
+        def test__calculates_correct_cliff_metrics(
+            self, mock_simulation_with_cliff_vars
+        ):
 
             task = object.__new__(GeneralEconomyTask)
             task.simulation = mock_simulation_with_cliff_vars