🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md
googleapis · Aug 24, 2023 · 3e66842 · 3e66842
1 parent dff8459
commit 3e66842
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/tests/unit/v1/test_aggregation.py b/tests/unit/v1/test_aggregation.py
@@ -611,7 +611,11 @@ def test_aggregation_from_query():
     kwargs = _helpers.make_retry_timeout_kwargs(retry, timeout)
 
     # Execute the query and check the response.
-    for aggregation_query in [query.count(alias="total"), query.sum("foo", alias="total"), query.avg("foo", alias="total")]:
+    for aggregation_query in [
+        query.count(alias="total"),
+        query.sum("foo", alias="total"),
+        query.avg("foo", alias="total"),
+    ]:
         # reset api mock
         firestore_api.run_aggregation_query.reset_mock()
         firestore_api.run_aggregation_query.return_value = iter([response_pb])

diff --git a/tests/unit/v1/test_async_aggregation.py b/tests/unit/v1/test_async_aggregation.py
@@ -456,7 +456,11 @@ async def test_async_aggregation_from_query():
     kwargs = _helpers.make_retry_timeout_kwargs(retry, timeout)
 
     # Execute each aggregation query type and check the response.
-    for aggregation_query in [query.count(alias="total"), query.sum("foo", alias="total"), query.avg("foo", alias="total")]:
+    for aggregation_query in [
+        query.count(alias="total"),
+        query.sum("foo", alias="total"),
+        query.avg("foo", alias="total"),
+    ]:
         # reset api mock
         firestore_api.run_aggregation_query.reset_mock()
         firestore_api.run_aggregation_query.return_value = iter([response_pb])