Skip to content

Commit 71f68bd

Browse files
committed
rebase master, update multistage test
1 parent 7fa7bc4 commit 71f68bd

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

tests/func/test_repro_multistage.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -261,25 +261,27 @@ def test_downstream(tmp_dir, dvc):
261261

262262
# B, C should be run (in any order) before D
263263
# See https://github.com/iterative/dvc/issues/3602
264-
evaluation = dvc.reproduce("Dvcfile:A-gen", downstream=True, force=True)
264+
evaluation = dvc.reproduce(
265+
PIPELINE_FILE + ":A-gen", downstream=True, force=True
266+
)
265267

266268
assert len(evaluation) == 5
267269
assert (
268270
isinstance(evaluation[0], PipelineStage)
269-
and evaluation[0].relpath == "Dvcfile"
271+
and evaluation[0].relpath == PIPELINE_FILE
270272
and evaluation[0].name == "A-gen"
271273
)
272274
names = set()
273275
for stage in evaluation[1:3]:
274276
if isinstance(stage, PipelineStage):
275-
assert stage.relpath == "Dvcfile"
277+
assert stage.relpath == PIPELINE_FILE
276278
names.add(stage.name)
277279
else:
278280
names.add(stage.relpath)
279281
assert names == {"B-gen", "C.dvc"}
280282
assert (
281283
isinstance(evaluation[3], PipelineStage)
282-
and evaluation[3].relpath == "Dvcfile"
284+
and evaluation[3].relpath == PIPELINE_FILE
283285
and evaluation[3].name == "D-gen"
284286
)
285287
assert (

0 commit comments

Comments
 (0)