Skip to content

Commit a1ceb1e

Browse files
committed
fix: ruff
1 parent 2aa1bc0 commit a1ceb1e

File tree

3 files changed

+7
-12
lines changed

3 files changed

+7
-12
lines changed

tests/api/answer/test_stream_processor.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -13,20 +13,18 @@
1313
- pre_fetch_docs
1414
"""
1515

16+
from unittest.mock import MagicMock, patch
17+
1618
import pytest
1719

20+
from application.api.answer.services.stream_processor import get_prompt
21+
1822
pytestmark = pytest.mark.skip(
1923
reason="Uses legacy Mongo ObjectId placeholder prompt IDs; get_prompt raises "
2024
"ValueError on the missing PG row. Needs prompt seeding via PG fixture or a "
2125
"monkeypatched get_prompt. Tracked as migration debt."
2226
)
2327

24-
from unittest.mock import MagicMock, patch
25-
26-
import pytest
27-
28-
from application.api.answer.services.stream_processor import get_prompt
29-
3028

3129
class TestGetPrompt:
3230

tests/test_coverage_gaps.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@
1111

1212
import pytest
1313

14-
from application.core.settings import settings
1514

1615

1716
# ---------------------------------------------------------------------------
@@ -471,7 +470,7 @@ def find_one_and_delete(self, q):
471470
key = (q.get("user_id"), q.get("tool_id"), q.get("todo_id"))
472471
return self.docs.pop(key, None)
473472

474-
fc = FakeCollection()
473+
FakeCollection()
475474
from application.agents.tools.todo_list import TodoListTool
476475

477476
return TodoListTool({"tool_id": "tt"}, user_id="u1")
@@ -599,7 +598,7 @@ def find_one_and_delete(self, q):
599598
key = f"{q.get('user_id')}:{q.get('tool_id')}"
600599
return self.docs.pop(key, None)
601600

602-
fc = FakeCollection()
601+
FakeCollection()
603602
from application.agents.tools.notes import NotesTool
604603

605604
return NotesTool({"tool_id": "nt"}, user_id="u1")

tests/test_usage.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import sys
2+
from contextlib import contextmanager
23

34
import pytest
45

@@ -290,9 +291,6 @@ def insert(self, **kwargs):
290291
self.inserted.append(kwargs)
291292

292293

293-
from contextlib import contextmanager
294-
295-
296294
@contextmanager
297295
def _fake_db_session():
298296
yield None

0 commit comments

Comments
 (0)