Update test_xw_consistency for current calibration config

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: baogorek

policyengine_us_data/tests/test_calibration/test_xw_consistency.py

@@ -48,7 +48,7 @@ def test_xw_matches_stacked_sim():
         prepare_base_sim_data,
     )
     from policyengine_us_data.utils.takeup import (
-        TAKEUP_AFFECTED_TARGETS,
+        SIMPLE_TAKEUP_VARS,
     )
 
     sim = Microsimulation(dataset=DATASET_PATH)
@@ -67,7 +67,6 @@ def test_xw_matches_stacked_sim():
 
     target_filter = {
         "variables": [
-            "aca_ptc",
             "snap",
             "household_count",
             "tax_unit_count",
@@ -77,18 +76,13 @@ def test_xw_matches_stacked_sim():
         geography=geography,
         sim=sim,
         target_filter=target_filter,
-        hierarchical_domains=["aca_ptc", "snap"],
+        hierarchical_domains=["snap"],
         rerandomize_takeup=True,
-        county_level=True,
+        county_level=False,
         workers=2,
     )
 
-    target_vars = set(target_filter["variables"])
-    takeup_filter = [
-        info["takeup_var"]
-        for key, info in TAKEUP_AFFECTED_TARGETS.items()
-        if key in target_vars
-    ]
+    takeup_filter = [spec["variable"] for spec in SIMPLE_TAKEUP_VARS]
 
     w = np.ones(n_total, dtype=np.float64)
     xw = X @ w
@@ -102,7 +96,7 @@ def test_xw_matches_stacked_sim():
         cd_weights[cd] = w[mask].sum()
     top_cds = sorted(cd_weights, key=cd_weights.get, reverse=True)[:N_CDS_TO_CHECK]
 
-    check_vars = ["aca_ptc", "snap"]
+    check_vars = ["snap"]
     tmpdir = tempfile.mkdtemp()
     base_data = prepare_base_sim_data(Path(DATASET_PATH))