Skip to content

Commit 49e8809

Browse files
committed
test(balatrobench): update for new strategy format
1 parent bdc9551 commit 49e8809

3 files changed

Lines changed: 4 additions & 2 deletions

File tree

tests/balatrobench/conftest.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@ def sample_strategy() -> Strategy:
3333
"""Sample Strategy instance."""
3434
return Strategy(
3535
name="Default",
36+
key="default",
3637
description="The default BalatroLLM strategy",
3738
author="BalatroLLM",
3839
version="1.0.0",

tests/balatrobench/unit/test_analyzer.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@ def simple_strategy() -> Strategy:
3232
"""Simple Strategy for local tests (distinct from conftest.sample_strategy)."""
3333
return Strategy(
3434
name="Default",
35+
key="default",
3536
description="Test strategy",
3637
author="Test",
3738
version="1.0.0",

tests/balatrobench/unit/test_writer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -502,9 +502,9 @@ def test_write_strategy_runs_creates_file(
502502

503503
result = writer.write_strategy_runs(sample_runs, "v1.0.0", "openai", "gpt-4o")
504504

505-
# Should return the path to written file
505+
# Should return the path to written file (uses strategy.key for path)
506506
expected_path = (
507-
tmp_path / "v1.0.0" / "openai" / "gpt-4o" / "Default" / "runs.json"
507+
tmp_path / "v1.0.0" / "openai" / "gpt-4o" / "default" / "runs.json"
508508
)
509509
assert result == expected_path
510510
assert result.exists()

0 commit comments

Comments
 (0)