xiaoxue_admin
/
mempalace
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update tests/test_hallways_pagination.py 

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
This commit is contained in:
Milla J 2026-06-02 19:56:16 -07:00 committed by GitHub
parent 4d3daf96d3
commit 526d58ae93
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/test_hallways_pagination.py
View File

@ -25,9 +25,13 @@ def _collection_that_rejects_where_get(drawers):
col.count.return_value = len(drawers)
def _get(limit=None, offset=0, include=None, where=None, ids=None, **kw):
if where is not None:
if where is not None and limit is None:
raise RuntimeError("Error executing plan: too many SQL variables")
page = drawers[offset : offset + limit] if limit is not None else drawers
filtered_drawers = drawers
if where and "wing" in where:
target_wing = where["wing"]
filtered_drawers = [d for d in drawers if isinstance(d, dict) and d.get("wing") == target_wing]
page = filtered_drawers[offset : offset + limit] if limit is not None else filtered_drawers
return {
"ids": [f"d{i}" for i in range(offset, offset + len(page))],
"metadatas": page,