fixed format

Author: daniel-sanche

packages/google-cloud-firestore/google/cloud/firestore_v1/pipeline_expressions.py

@@ -2958,7 +2958,7 @@ class DocumentMatches(BooleanExpression):
     """Creates a boolean expression for a document match query.
 
     Note: This Expression can only be used within a `Search` stage.
-    
+
     Example:
         >>> # Find documents matching the query string
         >>> DocumentMatches("search query")
@@ -2976,4 +2976,3 @@ def __init__(self, query: Expression | str):
             [Expression._cast_to_expr_or_convert_to_constant(query)],
             use_infix_repr=False,
         )
-

packages/google-cloud-firestore/google/cloud/firestore_v1/pipeline_stages.py

@@ -112,6 +112,7 @@ def percentage(value: float):
 
 class QueryEnhancement(Enum):
     """Define the query expansion behavior used by full-text search expressions."""
+
     DISABLED = "disabled"
     REQUIRED = "required"
     PREFERRED = "preferred"
@@ -157,10 +158,16 @@ def __init__(
         self.retrieval_depth = retrieval_depth
         self.sort = [sort] if isinstance(sort, Ordering) else sort
         self.add_fields = add_fields
-        self.select = [Field(s) if isinstance(s, str) else s for s in select] if select is not None else None
+        self.select = (
+            [Field(s) if isinstance(s, str) else s for s in select]
+            if select is not None
+            else None
+        )
         self.offset = offset
         self.query_enhancement = (
-            QueryEnhancement(query_enhancement.lower()) if isinstance(query_enhancement, str) else query_enhancement
+            QueryEnhancement(query_enhancement.lower())
+            if isinstance(query_enhancement, str)
+            else query_enhancement
         )
         self.language_code = language_code
 
@@ -516,17 +523,23 @@ def _pb_options(self) -> dict[str, Value]:
         if self.options.limit is not None:
             options["limit"] = Value(integer_value=self.options.limit)
         if self.options.retrieval_depth is not None:
-            options["retrieval_depth"] = Value(integer_value=self.options.retrieval_depth)
+            options["retrieval_depth"] = Value(
+                integer_value=self.options.retrieval_depth
+            )
         if self.options.sort is not None:
-            options["sort"] = Value(array_value={"values": [s._to_pb() for s in self.options.sort]})
+            options["sort"] = Value(
+                array_value={"values": [s._to_pb() for s in self.options.sort]}
+            )
         if self.options.add_fields is not None:
             options["add_fields"] = Selectable._to_value(self.options.add_fields)
         if self.options.select is not None:
             options["select"] = Selectable._to_value(self.options.select)
         if self.options.offset is not None:
             options["offset"] = Value(integer_value=self.options.offset)
         if self.options.query_enhancement is not None:
-            options["query_enhancement"] = Value(string_value=self.options.query_enhancement.value)
+            options["query_enhancement"] = Value(
+                string_value=self.options.query_enhancement.value
+            )
         if self.options.language_code is not None:
             options["language_code"] = Value(string_value=self.options.language_code)
         return options

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

@@ -33,7 +33,6 @@
 from google.cloud.firestore_v1 import pipeline_expressions as expr
 from google.cloud.firestore_v1 import pipeline_stages as stages
 from google.cloud.firestore_v1.vector import Vector
-from google.cloud.firestore_v1 import GeoPoint
 
 FIRESTORE_PROJECT = os.environ.get("GCLOUD_PROJECT")
 

packages/google-cloud-firestore/tests/unit/v1/test_pipeline_expressions.py

@@ -799,7 +799,10 @@ def test_between(self):
         assert len(instance.params) == 2
         assert instance.params[0].name == "greater_than_or_equal"
         assert instance.params[1].name == "less_than_or_equal"
-        assert repr(instance) == "And(Left.greater_than_or_equal(Lower), Left.less_than_or_equal(Upper))"
+        assert (
+            repr(instance)
+            == "And(Left.greater_than_or_equal(Lower), Left.less_than_or_equal(Upper))"
+        )
         infix_instance = arg1.between(arg2, arg3)
         assert infix_instance == instance
 

packages/google-cloud-firestore/tests/unit/v1/test_pipeline_stages.py

@@ -833,20 +833,22 @@ def test_search_string_query_wrapping(self):
         assert options.query.name == "document_matches"
 
     def test_search_query_enhancement_enum(self):
-        options = stages.SearchOptions(query="q", query_enhancement=stages.QueryEnhancement.REQUIRED)
+        options = stages.SearchOptions(
+            query="q", query_enhancement=stages.QueryEnhancement.REQUIRED
+        )
         assert options.query_enhancement == stages.QueryEnhancement.REQUIRED
-        
+
         stage = stages.Search(options)
         pb_opts = stage._pb_options()
         assert pb_opts["query_enhancement"].string_value == "required"
 
-
     def test_search_string_field_coercion(self):
         options = stages.SearchOptions(query="tech", select=["title"])
         assert len(options.select) == 1
         assert isinstance(options.select[0], Field)
         assert options.select[0].path == "title"
 
+
 class TestSelect:
     def _make_one(self, *args, **kwargs):
         return stages.Select(*args, **kwargs)