@@ -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