|
13 | 13 | LintStage, |
14 | 14 | PlanStage, |
15 | 15 | ReviewStage, |
16 | | - TestStage, |
| 16 | + TestingStage, |
17 | 17 | VerifyStage, |
18 | 18 | create_stage, |
19 | 19 | STAGE_CLASSES, |
@@ -182,7 +182,7 @@ def test_creates_review_stage(self): |
182 | 182 | def test_creates_test_stage(self): |
183 | 183 | config = _make_stage_config(name="test", prompt_file="prompts/test.txt") |
184 | 184 | stage = create_stage(config, _make_model()) |
185 | | - assert isinstance(stage, TestStage) |
| 185 | + assert isinstance(stage, TestingStage) |
186 | 186 |
|
187 | 187 | def test_creates_lint_stage(self): |
188 | 188 | config = _make_stage_config(name="lint", prompt_file="prompts/lint.txt") |
@@ -260,15 +260,15 @@ async def test_review_stage_prompt(self): |
260 | 260 |
|
261 | 261 | async def test_test_stage_prompt(self): |
262 | 262 | config = _make_stage_config(name="test", prompt_file="prompts/test.txt") |
263 | | - stage = TestStage(config=config, model_config=_make_model(), prompts_dir="/nonexistent") |
| 263 | + stage = TestingStage(config=config, model_config=_make_model(), prompts_dir="/nonexistent") |
264 | 264 | state = PipelineState(user_request="test") |
265 | 265 | prompt = await stage.build_prompt(state) |
266 | 266 | assert "test suite" in prompt |
267 | 267 | assert "PLAN.md" in prompt |
268 | 268 |
|
269 | 269 | async def test_test_stage_with_feedback(self): |
270 | 270 | config = _make_stage_config(name="test", prompt_file="prompts/test.txt") |
271 | | - stage = TestStage(config=config, model_config=_make_model(), prompts_dir="/nonexistent") |
| 271 | + stage = TestingStage(config=config, model_config=_make_model(), prompts_dir="/nonexistent") |
272 | 272 | state = PipelineState( |
273 | 273 | user_request="test", |
274 | 274 | review_feedback="3 tests failing", |
|
0 commit comments