Format test files with black

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>

Author: baogorek

policyengine_us_data/tests/test_local_area_calibration/create_test_fixture.py

@@ -23,12 +23,16 @@ def create_test_fixture():
     np.random.seed(SEED)
 
     # Create household structure: 50 households with 1-4 persons each
-    persons_per_hh = np.random.choice([1, 2, 3, 4], size=N_HOUSEHOLDS, p=[0.3, 0.4, 0.2, 0.1])
+    persons_per_hh = np.random.choice(
+        [1, 2, 3, 4], size=N_HOUSEHOLDS, p=[0.3, 0.4, 0.2, 0.1]
+    )
     n_persons = persons_per_hh.sum()
 
     # Household-level arrays
     household_ids = np.arange(N_HOUSEHOLDS, dtype=np.int32)
-    household_weights = np.random.uniform(500, 3000, N_HOUSEHOLDS).astype(np.float32)
+    household_weights = np.random.uniform(500, 3000, N_HOUSEHOLDS).astype(
+        np.float32
+    )
 
     # Assign households to states (use NC=37 and AK=2 for testing)
     # 40 households in NC, 10 in AK
@@ -53,7 +57,7 @@ def create_test_fixture():
     ages = np.random.choice(
         [5, 15, 25, 35, 45, 55, 65, 75],
         size=n_persons,
-        p=[0.1, 0.1, 0.15, 0.2, 0.15, 0.15, 0.1, 0.05]
+        p=[0.1, 0.1, 0.15, 0.2, 0.15, 0.15, 0.1, 0.05],
     ).astype(np.int32)
 
     # Employment income: working-age adults have income
@@ -98,14 +102,18 @@ def create_test_fixture():
         f["household_id"].create_dataset(TIME_PERIOD, data=household_ids)
 
         f.create_group("household_weight")
-        f["household_weight"].create_dataset(TIME_PERIOD, data=household_weights)
+        f["household_weight"].create_dataset(
+            TIME_PERIOD, data=household_weights
+        )
 
         # Person variables
         f.create_group("person_id")
         f["person_id"].create_dataset(TIME_PERIOD, data=person_ids)
 
         f.create_group("person_household_id")
-        f["person_household_id"].create_dataset(TIME_PERIOD, data=person_household_ids)
+        f["person_household_id"].create_dataset(
+            TIME_PERIOD, data=person_household_ids
+        )
 
         f.create_group("person_weight")
         f["person_weight"].create_dataset(TIME_PERIOD, data=person_weights)
@@ -114,14 +122,18 @@ def create_test_fixture():
         f["age"].create_dataset(TIME_PERIOD, data=ages)
 
         f.create_group("employment_income")
-        f["employment_income"].create_dataset(TIME_PERIOD, data=employment_income)
+        f["employment_income"].create_dataset(
+            TIME_PERIOD, data=employment_income
+        )
 
         # Tax unit
         f.create_group("tax_unit_id")
         f["tax_unit_id"].create_dataset(TIME_PERIOD, data=tax_unit_ids)
 
         f.create_group("person_tax_unit_id")
-        f["person_tax_unit_id"].create_dataset(TIME_PERIOD, data=person_tax_unit_ids)
+        f["person_tax_unit_id"].create_dataset(
+            TIME_PERIOD, data=person_tax_unit_ids
+        )
 
         f.create_group("tax_unit_weight")
         f["tax_unit_weight"].create_dataset(TIME_PERIOD, data=tax_unit_weights)
@@ -131,7 +143,9 @@ def create_test_fixture():
         f["spm_unit_id"].create_dataset(TIME_PERIOD, data=spm_unit_ids)
 
         f.create_group("person_spm_unit_id")
-        f["person_spm_unit_id"].create_dataset(TIME_PERIOD, data=person_spm_unit_ids)
+        f["person_spm_unit_id"].create_dataset(
+            TIME_PERIOD, data=person_spm_unit_ids
+        )
 
         f.create_group("spm_unit_weight")
         f["spm_unit_weight"].create_dataset(TIME_PERIOD, data=spm_unit_weights)
@@ -141,7 +155,9 @@ def create_test_fixture():
         f["family_id"].create_dataset(TIME_PERIOD, data=family_ids)
 
         f.create_group("person_family_id")
-        f["person_family_id"].create_dataset(TIME_PERIOD, data=person_family_ids)
+        f["person_family_id"].create_dataset(
+            TIME_PERIOD, data=person_family_ids
+        )
 
         f.create_group("family_weight")
         f["family_weight"].create_dataset(TIME_PERIOD, data=family_weights)
@@ -151,10 +167,14 @@ def create_test_fixture():
         f["marital_unit_id"].create_dataset(TIME_PERIOD, data=marital_unit_ids)
 
         f.create_group("person_marital_unit_id")
-        f["person_marital_unit_id"].create_dataset(TIME_PERIOD, data=person_marital_unit_ids)
+        f["person_marital_unit_id"].create_dataset(
+            TIME_PERIOD, data=person_marital_unit_ids
+        )
 
         f.create_group("marital_unit_weight")
-        f["marital_unit_weight"].create_dataset(TIME_PERIOD, data=marital_unit_weights)
+        f["marital_unit_weight"].create_dataset(
+            TIME_PERIOD, data=marital_unit_weights
+        )
 
         # Geography (household level)
         f.create_group("state_fips")

policyengine_us_data/tests/test_local_area_calibration/test_stacked_dataset_builder.py

@@ -11,9 +11,7 @@
     create_sparse_cd_stacked_dataset,
 )
 
-FIXTURE_PATH = (
-    "policyengine_us_data/tests/test_local_area_calibration/test_fixture_50hh.h5"
-)
+FIXTURE_PATH = "policyengine_us_data/tests/test_local_area_calibration/test_fixture_50hh.h5"
 TEST_CDS = ["3701", "201"]  # NC-01 and AK at-large
 SEED = 42
 
@@ -151,7 +149,9 @@ def test_counties_match_state(self, stacked_result):
                     "_AK"
                 ), f"AK county should end with _AK: {county}"
 
-    def test_household_count_matches_weights(self, stacked_result, test_weights):
+    def test_household_count_matches_weights(
+        self, stacked_result, test_weights
+    ):
         """Number of output households should match non-zero weights."""
         hh_df = stacked_result["hh_df"]
         expected_households = (test_weights > 0).sum()