copy over to async

Author: geoffrey-wu

qbreader/asynchronous.py

@@ -80,6 +80,8 @@ async def query(
         maxReturnLength: Optional[int] = 25,
         tossupPagination: Optional[int] = 1,
         bonusPagination: Optional[int] = 1,
+        min_year: int = Year.MIN_YEAR,
+        max_year: int = Year.CURRENT_YEAR,
     ) -> QueryResponse:
         """Query the qbreader database for questions.
 
@@ -127,7 +129,10 @@ class type.
             The page of tossups to return.
         bonusPagination : int, default = 1
             The page of bonuses to return.
-
+        min_year : int, default = Year.MIN_YEAR
+            The earliest year to search.
+        max_year : int, default = Year.CURRENT_YEAR
+            The latest year to search.
         Returns
         -------
         QueryResponse
@@ -184,6 +189,15 @@ class type.
             elif param < 1:
                 raise ValueError(f"{name} must be at least 1.")
 
+        for name, year in {
+            "minYear": min_year,
+            "maxYear": max_year,
+        }.items():
+            if not isinstance(year, int):
+                raise TypeError(
+                    f"{name} must be an integer, not {type(param).__name__}."
+                )
+
         url = BASE_URL + "/query"
 
         (
@@ -209,6 +223,8 @@ class type.
             "maxReturnLength": maxReturnLength,
             "tossupPagination": tossupPagination,
             "bonusPagination": bonusPagination,
+            "minYear": min_year,
+            "maxYear": max_year,
         }
         data = api_utils.prune_none(data)
 

tests/test_async.py

@@ -116,6 +116,22 @@ async def test_query(self, qbr, params: dict[str, Any], expected_answer: str):
             )
             assert judgement.correct()
 
+    @pytest.mark.asyncio
+    async def test_query_min_max_year_range(self, qbr):
+        min_year = 2010
+        max_year = 2015
+
+        query = await qbr.query(
+            questionType="tossup",
+            searchType="question",
+            min_year=min_year,
+            max_year=max_year,
+            maxReturnLength=10,
+        )
+
+        for tossup in query.tossups:
+            assert min_year <= tossup.set.year <= max_year
+
     @pytest.mark.asyncio
     @pytest.mark.parametrize(
         "params, exception",