Skip to content

Commit 84f64f6

Browse files
committed
fix(ci): Update test_stage.py to use renamed TestingStage class
1 parent eb4f7da commit 84f64f6

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

pipeline/tests/test_stage.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
LintStage,
1414
PlanStage,
1515
ReviewStage,
16-
TestStage,
16+
TestingStage,
1717
VerifyStage,
1818
create_stage,
1919
STAGE_CLASSES,
@@ -182,7 +182,7 @@ def test_creates_review_stage(self):
182182
def test_creates_test_stage(self):
183183
config = _make_stage_config(name="test", prompt_file="prompts/test.txt")
184184
stage = create_stage(config, _make_model())
185-
assert isinstance(stage, TestStage)
185+
assert isinstance(stage, TestingStage)
186186

187187
def test_creates_lint_stage(self):
188188
config = _make_stage_config(name="lint", prompt_file="prompts/lint.txt")
@@ -260,15 +260,15 @@ async def test_review_stage_prompt(self):
260260

261261
async def test_test_stage_prompt(self):
262262
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")
264264
state = PipelineState(user_request="test")
265265
prompt = await stage.build_prompt(state)
266266
assert "test suite" in prompt
267267
assert "PLAN.md" in prompt
268268

269269
async def test_test_stage_with_feedback(self):
270270
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")
272272
state = PipelineState(
273273
user_request="test",
274274
review_feedback="3 tests failing",

0 commit comments

Comments
 (0)