Skip to content

Commit 7fe4be6

Browse files
committed
fixed lint
1 parent 700eed6 commit 7fe4be6

File tree

1 file changed

+19
-10
lines changed

1 file changed

+19
-10
lines changed

packages/google-cloud-firestore/tests/system/test_pipeline_acceptance.py

Lines changed: 19 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -133,25 +133,30 @@ def test_pipeline_expected_errors(test_dict, client):
133133
assert match, f"error '{found_error}' does not match '{error_regex}'"
134134

135135

136-
137-
def _assert_pipeline_results(got_results, expected_results, expected_approximate_results, expected_count):
136+
def _assert_pipeline_results(
137+
got_results, expected_results, expected_approximate_results, expected_count
138+
):
138139
if expected_results:
139140
assert got_results == expected_results
140141
if expected_approximate_results is not None:
141142
tolerance = 1e-4
142-
if isinstance(expected_approximate_results, dict) and "data" in expected_approximate_results:
143-
if "config" in expected_approximate_results and "absolute_tolerance" in expected_approximate_results["config"]:
143+
if (
144+
isinstance(expected_approximate_results, dict)
145+
and "data" in expected_approximate_results
146+
):
147+
if (
148+
"config" in expected_approximate_results
149+
and "absolute_tolerance" in expected_approximate_results["config"]
150+
):
144151
tolerance = expected_approximate_results["config"]["absolute_tolerance"]
145152
expected_approximate_results = expected_approximate_results["data"]
146-
153+
147154
assert len(got_results) == len(expected_approximate_results), (
148155
"got unexpected result count"
149156
)
150157
for idx in range(len(got_results)):
151158
expected = expected_approximate_results[idx]
152-
assert got_results[idx] == pytest.approx(
153-
expected, abs=tolerance
154-
)
159+
assert got_results[idx] == pytest.approx(expected, abs=tolerance)
155160
if expected_count is not None:
156161
assert len(got_results) == expected_count
157162

@@ -181,7 +186,9 @@ def test_pipeline_results(test_dict, client):
181186
pipeline = parse_pipeline(client, test_dict["pipeline"])
182187
# check if server responds as expected
183188
got_results = [snapshot.data() for snapshot in pipeline.stream()]
184-
_assert_pipeline_results(got_results, expected_results, expected_approximate_results, expected_count)
189+
_assert_pipeline_results(
190+
got_results, expected_results, expected_approximate_results, expected_count
191+
)
185192
if expected_end_state:
186193
for doc_path, expected_content in expected_end_state.items():
187194
doc_ref = client.document(doc_path)
@@ -243,7 +250,9 @@ async def test_pipeline_results_async(test_dict, async_client):
243250
pipeline = parse_pipeline(async_client, test_dict["pipeline"])
244251
# check if server responds as expected
245252
got_results = [snapshot.data() async for snapshot in pipeline.stream()]
246-
_assert_pipeline_results(got_results, expected_results, expected_approximate_results, expected_count)
253+
_assert_pipeline_results(
254+
got_results, expected_results, expected_approximate_results, expected_count
255+
)
247256
if expected_end_state:
248257
for doc_path, expected_content in expected_end_state.items():
249258
doc_ref = async_client.document(doc_path)

0 commit comments

Comments
 (0)