chore: ran linting

This commit is contained in:
Anish Sarkar 2026-02-26 01:29:25 +05:30
parent 380c1c3877
commit 25df3dff64
3 changed files with 17 additions and 10 deletions

View file

@ -175,7 +175,10 @@ class TestPageLimitNotification:
)
latest = notifications[0]
assert "page limit" in latest["title"].lower() or "page limit" in latest["message"].lower(), (
assert (
"page limit" in latest["title"].lower()
or "page limit" in latest["message"].lower()
), (
f"Notification should mention page limit: title={latest['title']!r}, "
f"message={latest['message']!r}"
)
@ -217,7 +220,11 @@ class TestDocumentProcessingNotification:
type_filter="document_processing",
search_space_id=search_space_id,
)
completed = [n for n in notifications if n.get("metadata", {}).get("processing_stage") == "completed"]
completed = [
n
for n in notifications
if n.get("metadata", {}).get("processing_stage") == "completed"
]
assert len(completed) >= 1, (
"Expected at least one document_processing notification with 'completed' stage"
)
@ -257,9 +264,7 @@ class TestPagesUnchangedOnProcessingFailure:
assert statuses[did]["status"]["state"] == "failed"
used, _ = await page_limits.get()
assert used == 10, (
f"pages_used should remain 10 after ETL failure, got {used}"
)
assert used == 10, f"pages_used should remain 10 after ETL failure, got {used}"
# ---------------------------------------------------------------------------

View file

@ -130,7 +130,10 @@ class TestTotalSizeLimit:
):
chunk_size = 45 * 1024 * 1024 # 45 MB each
files = [
("files", (f"chunk_{i}.txt", io.BytesIO(b"\x00" * chunk_size), "text/plain"))
(
"files",
(f"chunk_{i}.txt", io.BytesIO(b"\x00" * chunk_size), "text/plain"),
)
for i in range(5) # 5 x 45 MB = 225 MB > 200 MB
]
resp = await client.post(