improve coverage

Author: WaVEV

django_mongodb_backend/expressions/search.py

@@ -989,7 +989,7 @@ def as_mql(self, compiler, connection):
         if self.exact:
             params["exact"] = self.exact.value
         if self.filter:
-            params["filter"] = self.filter.as_mql(compiler, connection)
+            params["filter"] = self.filter.value
         return {"$vectorSearch": params}
 
 

tests/atlas_search_/test_search.py

@@ -980,7 +980,11 @@ def setUpClass(cls):
                         "numDimensions": 3,
                         "similarity": "cosine",
                         "quantization": "scalar",
-                    }
+                    },
+                    {
+                        "type": "filter",
+                        "path": "number",
+                    },
                 ]
             },
             type="vectorSearch",
@@ -999,14 +1003,22 @@ def setUp(self):
             body="This article is about pasta",
             plot_embedding=[0.9, 0.8, 0.7],
         )
+        Article.objects.create(
+            headline="Local team wins championship",
+            number=3,
+            body="This article is about sports",
+            plot_embedding=[-0.1, 0.7, 0.7],
+        )
 
     def test_vector_search(self):
         vector_query = [0.1, 0.2, 0.3]
         expr = SearchVector(
             path="plot_embedding",
             query_vector=vector_query,
             num_candidates=5,
-            limit=2,
+            limit=3,
+            filter={"number": {"$lt": 3}},
+            exact=False,
         )
         qs = Article.objects.annotate(score=expr).order_by("-score")
         self.assertCountEqual(qs, [self.mars, self.cooking])