Skip to content

Commit b381f01

Browse files
committed
Format SOI refresh files for lint
1 parent a4abb52 commit b381f01

2 files changed

Lines changed: 15 additions & 5 deletions

File tree

policyengine_us_data/storage/calibration_targets/refresh_soi_table_targets.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -165,7 +165,9 @@ def _validate_source_year(all_targets: pd.DataFrame, source_year: int) -> None:
165165
drop=True
166166
)
167167

168-
pd.testing.assert_frame_equal(expected, actual, check_dtype=False, check_exact=False)
168+
pd.testing.assert_frame_equal(
169+
expected, actual, check_dtype=False, check_exact=False
170+
)
169171

170172

171173
def _serialize_bound(value: float) -> str:

tests/test_refresh_soi_table_targets.py

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,9 @@ def test_build_target_year_rows_reads_standard_table_cells(monkeypatch):
9393

9494
monkeypatch.setattr(module, "_load_workbook", lambda table, year: workbook)
9595

96-
refreshed = module.build_target_year_rows(targets, source_year=2021, target_year=2023)
96+
refreshed = module.build_target_year_rows(
97+
targets, source_year=2021, target_year=2023
98+
)
9799

98100
assert refreshed["Year"].tolist() == [2023, 2023]
99101
assert refreshed["Value"].tolist() == [123_000.0, 789.0]
@@ -157,12 +159,16 @@ def test_build_target_year_rows_sums_partnership_and_s_corp_components(monkeypat
157159

158160
monkeypatch.setattr(module, "_load_workbook", lambda table, year: workbook)
159161

160-
refreshed = module.build_target_year_rows(targets, source_year=2021, target_year=2023)
162+
refreshed = module.build_target_year_rows(
163+
targets, source_year=2021, target_year=2023
164+
)
161165

162166
assert refreshed["Value"].tolist() == [30.0, 70_000.0, 11.0, 15_000.0]
163167

164168

165-
def test_build_target_year_rows_differences_top_tail_rows_and_updates_bounds(monkeypatch):
169+
def test_build_target_year_rows_differences_top_tail_rows_and_updates_bounds(
170+
monkeypatch,
171+
):
166172
module = load_module()
167173
workbook = make_workbook(cols=25)
168174

@@ -226,7 +232,9 @@ def test_build_target_year_rows_differences_top_tail_rows_and_updates_bounds(mon
226232

227233
monkeypatch.setattr(module, "_load_workbook", lambda table, year: workbook)
228234

229-
refreshed = module.build_target_year_rows(targets, source_year=2021, target_year=2023)
235+
refreshed = module.build_target_year_rows(
236+
targets, source_year=2021, target_year=2023
237+
)
230238

231239
assert refreshed["Value"].tolist() == [10.0, 50.0, 200_000.0, 800_000.0]
232240
assert refreshed["AGI lower bound"].tolist() == [1_000.0, 400.0, 1_000.0, 400.0]

0 commit comments

Comments
 (0)