Skip to content

Commit 40c814c

Browse files
committed
chore: Format
1 parent f3a59a8 commit 40c814c

6 files changed

Lines changed: 21 additions & 18 deletions

File tree

projects/policyengine-api-simulation/src/modal/app.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -104,5 +104,3 @@ def run_simulation(params: dict) -> dict:
104104
return result
105105
finally:
106106
logfire.force_flush()
107-
108-

projects/policyengine-api-simulation/tests/gateway/test_endpoints.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -99,9 +99,7 @@ def test__given_unknown_version__then_raises_value_error(self, mock_modal):
9999
from src.modal.gateway.endpoints import get_app_name
100100

101101
# Given
102-
mock_modal["dicts"]["simulation-api-us-versions"] = {
103-
"1.459.0": "some-app"
104-
}
102+
mock_modal["dicts"]["simulation-api-us-versions"] = {"1.459.0": "some-app"}
105103

106104
# When / Then
107105
with pytest.raises(ValueError, match="Unknown version"):

projects/policyengine-api-simulation/tests/test_logfire_integration.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -152,9 +152,7 @@ def test_span_receives_input_params(self, mock_logfire, mock_span):
152152
pass
153153

154154
# Then
155-
mock_logfire.span.assert_called_once_with(
156-
"run_simulation", input_params=params
157-
)
155+
mock_logfire.span.assert_called_once_with("run_simulation", input_params=params)
158156

159157
def test_span_captures_output_result(self, mock_logfire, mock_span):
160158
"""

projects/policyengine-api-simulation/tests/test_modal_scripts.py

Lines changed: 15 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -138,9 +138,12 @@ def test_generates_success_summary(self, temp_github_step_summary):
138138

139139
result = subprocess.run(
140140
[
141-
"bash", str(self.script),
142-
"success", "https://beta.example.com",
143-
"success", "https://prod.example.com",
141+
"bash",
142+
str(self.script),
143+
"success",
144+
"https://beta.example.com",
145+
"success",
146+
"https://prod.example.com",
144147
],
145148
capture_output=True,
146149
text=True,
@@ -165,9 +168,12 @@ def test_generates_skipped_summary(self, temp_github_step_summary):
165168

166169
result = subprocess.run(
167170
[
168-
"bash", str(self.script),
169-
"skipped", "",
170-
"success", "https://prod.example.com",
171+
"bash",
172+
str(self.script),
173+
"skipped",
174+
"",
175+
"success",
176+
"https://prod.example.com",
171177
],
172178
capture_output=True,
173179
text=True,
@@ -360,4 +366,6 @@ def test_all_scripts_have_valid_syntax(self, all_modal_scripts):
360366
capture_output=True,
361367
text=True,
362368
)
363-
assert result.returncode == 0, f"{script.name} has syntax errors: {result.stderr}"
369+
assert result.returncode == 0, (
370+
f"{script.name} has syntax errors: {result.stderr}"
371+
)

projects/policyengine-api-simulation/tests/test_pandas3_compatibility.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
"""Test pandas 3.0 compatibility with enum encoding."""
2+
23
import pandas as pd
34
from policyengine_core.enums import Enum
45

@@ -12,8 +13,8 @@ def test_enum_encode_with_pandas_series():
1213
"""Test that Enum.encode works with pandas Series."""
1314
enum_items = [SampleEnum.VALUE_A, SampleEnum.VALUE_B, SampleEnum.VALUE_A]
1415
series = pd.Series(enum_items)
15-
16+
1617
encoded = SampleEnum.encode(series)
17-
18+
1819
assert len(encoded) == 3
1920
assert list(encoded) == [0, 1, 0]

projects/policyengine-api-simulation/tests/test_placeholder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,4 +3,4 @@
33

44
def test_placeholder():
55
"""Placeholder test that always passes."""
6-
assert True
6+
assert True

0 commit comments

Comments
 (0)