Skip to content

Commit 58f5f9d

Browse files
committed
fix the remaining test
1 parent 8dadf8b commit 58f5f9d

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

tests/core/test_plan_evaluator.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -69,10 +69,12 @@ def test_builtin_evaluator_push(sushi_context: Context, make_snapshot):
6969
stages = plan_stages.build_plan_stages(
7070
evaluatable_plan, sushi_context.state_sync, sushi_context.default_catalog
7171
)
72-
assert isinstance(stages[0], plan_stages.CreateSnapshotRecordsStage)
73-
evaluator.visit_create_snapshot_records_stage(stages[0], evaluatable_plan)
74-
assert isinstance(stages[1], plan_stages.PhysicalLayerUpdateStage)
75-
evaluator.visit_physical_layer_update_stage(stages[1], evaluatable_plan)
72+
assert isinstance(stages[1], plan_stages.CreateSnapshotRecordsStage)
73+
evaluator.visit_create_snapshot_records_stage(stages[1], evaluatable_plan)
74+
assert isinstance(stages[2], plan_stages.PhysicalLayerSchemaCreationStage)
75+
evaluator.visit_physical_layer_schema_creation_stage(stages[2], evaluatable_plan)
76+
assert isinstance(stages[3], plan_stages.BackfillStage)
77+
evaluator.visit_backfill_stage(stages[3], evaluatable_plan)
7678

7779
assert (
7880
len(sushi_context.state_sync.get_snapshots([new_model_snapshot, new_view_model_snapshot]))

0 commit comments

Comments
 (0)