|
30 | 30 | test_table: list[pytest_helper.TestTableItem] = [ |
31 | 31 | pytest_helper.TestTableItem( |
32 | 32 | name='test_tuned_models', |
33 | | - parameters=types._ListModelsParameters(config={'query_base': False}), |
| 33 | + parameters=types._ListModelsParameters( |
| 34 | + config={'query_base': False, 'page_size': 3} |
| 35 | + ), |
| 36 | + ), |
| 37 | + pytest_helper.TestTableItem( |
| 38 | + name='test_tuned_models_with_filter', |
| 39 | + parameters=types._ListModelsParameters( |
| 40 | + config={'query_base': False, |
| 41 | + 'page_size': 3, |
| 42 | + 'filter': 'displayName="gemini-2.5-flash-1b0689e3-9773-43b4-97eb-c8140d5f183b"'} |
| 43 | + ), |
34 | 44 | ), |
35 | 45 | pytest_helper.TestTableItem( |
36 | 46 | name='test_base_models', |
@@ -193,13 +203,13 @@ def test_empty_api_response_empty_dict_headers(mock_api_client, client): |
193 | 203 |
|
194 | 204 | @pytest.mark.asyncio |
195 | 205 | async def test_tuned_models_async_pager(client): |
196 | | - pager = await client.aio.models.list(config={'page_size': 10, 'query_base': False}) |
| 206 | + pager = await client.aio.models.list(config={'page_size': 3, 'query_base': False}) |
197 | 207 |
|
198 | 208 | assert 'Content-Type' in pager.sdk_http_response.headers |
199 | 209 | assert 'Content-Encoding' in pager.sdk_http_response.headers |
200 | 210 | assert pager.name == 'models' |
201 | | - assert pager.page_size == 10 |
202 | | - assert len(pager) <= 10 |
| 211 | + assert pager.page_size == 3 |
| 212 | + assert len(pager) <= 3 |
203 | 213 |
|
204 | 214 | # Iterate through all the pages. Then next_page() should raise an exception. |
205 | 215 | async for _ in pager: |
|
0 commit comments