Skip to content

Commit 50748fe

Browse files
committed
fix: resolve ruff lint issues in test files
1 parent b809fcb commit 50748fe

3 files changed

Lines changed: 10 additions & 6 deletions

File tree

tests/test_context_window.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,25 +1,26 @@
11
from hawk.context_window import ContextOptimizer
22
from hawk.types import Message
33

4+
45
def test_context_optimizer_compaction():
56
optimizer = ContextOptimizer(max_tokens=20) # Approx 80 chars
6-
7+
78
# 3 messages, ~30 chars each
89
messages = [
910
Message(role="user", content="This is message number one."),
1011
Message(role="user", content="This is message number two."),
1112
Message(role="user", content="This is message number three.")
1213
]
13-
14+
1415
# "Sys" = 3 chars = 0 tokens (3//4). Available = 20
1516
# msg3: 29 chars = 7 tokens. Current=7
1617
# msg2: 27 chars = 6 tokens. Current=13
1718
# msg1: 27 chars = 6 tokens. Current=19
1819
# Wait, my manual math shows they all fit. Let's make max_tokens smaller.
19-
20+
2021
optimizer = ContextOptimizer(max_tokens=15)
2122
compacted = optimizer.compact_messages(messages, system_prompt="Sys")
22-
23+
2324
assert len(compacted) == 2
2425
assert compacted[0].content == "This is message number two."
2526
assert compacted[1].content == "This is message number three."

tests/test_graph_dsl.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,13 @@
11
from hawk.graph_dsl import GraphQuery, NodeType
22

3+
34
def test_graph_query_shortest_path():
45
g = GraphQuery()
56
g.node("A", NodeType.START).node("B").node("C").node("D", NodeType.END)
67
g.edge("A", "B").edge("B", "C").edge("C", "D").edge("A", "D")
7-
8+
89
path = g.shortest_path("A", "D")
910
assert path == ["A", "D"]
10-
11+
1112
path_long = g.shortest_path("A", "C")
1213
assert path_long == ["A", "B", "C"]

tests/test_harness.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
11
import pytest
2+
23
from hawk.harness import TestHarness
34

5+
46
@pytest.mark.asyncio
57
async def test_harness_validators():
68
harness = TestHarness("test")

0 commit comments

Comments
 (0)