Add pagination tests for limits exceeded (#737)

marqo-ai · Jan 25, 2024 · 8697cdf · 8697cdf
1 parent e9ea2c8
commit 8697cdf
Showing 1 changed file with 26 additions and 3 deletions.
diff --git a/tests/tensor_search/test_pagination.py b/tests/tensor_search/test_pagination.py
@@ -6,6 +6,7 @@
 
 import requests
 
+from marqo.api import exceptions as api_exceptions
 from marqo.api.exceptions import (
     IllegalRequestedDocCount
 )
@@ -165,17 +166,39 @@ def read_var(var):
 
                     self.assertTrue(max_offset in offsets_covered, "Max offset not covered. Check test parameters")
 
-    def test_pagination_limt_exceeded_error(self):
+    def test_pagination_limit_exceeded_error(self):
         """
         Verify InvalidArgs error is raised when limit is exceeded
         """
-        self.fail()
+        for index in [self.index_structured, self.index_unstructured]:
+            for search_method in [SearchMethod.TENSOR, SearchMethod.LEXICAL]:
+                for limit in [1001, 2000, 10000]:
+                    with self.subTest(f'Index: {index.type}, Search method: {search_method}, Limit: {limit}'):
+                        with self.assertRaises(api_exceptions.IllegalRequestedDocCount):
+                            tensor_search.search(
+                                search_method=search_method,
+                                config=self.config,
+                                index_name=index.name,
+                                text='my title',
+                                result_count=limit, offset=0,
+                            )
 
     def test_pagination_offset_exceeded_error(self):
         """
         Verify InvalidArgs error is raised when offset is exceeded
         """
-        self.fail()
+        for index in [self.index_structured, self.index_unstructured]:
+            for search_method in [SearchMethod.TENSOR, SearchMethod.LEXICAL]:
+                for offset in [10001, 12000, 50000]:
+                    with self.subTest(f'Index: {index.type}, Search method: {search_method}, Offset: {offset}'):
+                        with self.assertRaises(api_exceptions.IllegalRequestedDocCount):
+                            tensor_search.search(
+                                search_method=search_method,
+                                config=self.config,
+                                index_name=index.name,
+                                text='my title',
+                                result_count=100, offset=offset,
+                            )
 
     @unittest.skip
     def test_pagination_multi_field(self):