@@ -1004,7 +1004,7 @@ def task_example(
1004
1004
1005
1005
1006
1006
@pytest .mark .end_to_end ()
1007
- def test_task_that_depends_on_delayed_task (tmp_path ):
1007
+ def test_task_that_depends_on_delayed_task (runner , tmp_path ):
1008
1008
source = """
1009
1009
from typing_extensions import Annotated
1010
1010
from pytask import DirectoryNode, task
@@ -1024,12 +1024,10 @@ def task_depends(
1024
1024
"""
1025
1025
tmp_path .joinpath ("task_module.py" ).write_text (textwrap .dedent (source ))
1026
1026
1027
- session = build (paths = tmp_path )
1028
-
1029
- assert session .exit_code == ExitCode .OK
1030
- assert len (session .tasks ) == 2
1031
- assert len (session .tasks [0 ].produces ["return" ]) == 2
1032
- assert len (session .tasks [1 ].depends_on ["paths" ]) == 2
1027
+ result = runner .invoke (cli , [tmp_path .as_posix ()])
1028
+ assert result .exit_code == ExitCode .OK
1029
+ assert "2 Collected tasks" in result .output
1030
+ assert "2 Succeeded" in result .output
1033
1031
1034
1032
1035
1033
@pytest .mark .end_to_end ()
@@ -1062,7 +1060,7 @@ def task_depends(
1062
1060
1063
1061
1064
1062
@pytest .mark .end_to_end ()
1065
- def test_delayed_task_generation (tmp_path ):
1063
+ def test_delayed_task_generation (runner , tmp_path ):
1066
1064
source = """
1067
1065
from typing_extensions import Annotated
1068
1066
from pytask import DirectoryNode, task
@@ -1087,18 +1085,16 @@ def task_copy(
1087
1085
"""
1088
1086
tmp_path .joinpath ("task_module.py" ).write_text (textwrap .dedent (source ))
1089
1087
1090
- session = build (paths = tmp_path )
1091
-
1092
- assert session .exit_code == ExitCode .OK
1093
- assert len (session .tasks ) == 4
1094
- assert len (session .tasks [0 ].produces ["return" ]) == 2
1095
- assert len (session .tasks [1 ].depends_on ["paths" ]) == 2
1088
+ result = runner .invoke (cli , [tmp_path .as_posix ()])
1089
+ assert result .exit_code == ExitCode .OK
1090
+ assert "4 Collected tasks" in result .output
1091
+ assert "4 Succeeded" in result .output
1096
1092
assert tmp_path .joinpath ("a-copy.txt" ).exists ()
1097
1093
assert tmp_path .joinpath ("b-copy.txt" ).exists ()
1098
1094
1099
1095
1100
1096
@pytest .mark .end_to_end ()
1101
- def test_delayed_task_generation_with_generator (tmp_path ):
1097
+ def test_delayed_task_generation_with_generator (runner , tmp_path ):
1102
1098
source = """
1103
1099
from typing_extensions import Annotated
1104
1100
from pytask import DirectoryNode, task
@@ -1125,18 +1121,16 @@ def task_copy(
1125
1121
"""
1126
1122
tmp_path .joinpath ("task_module.py" ).write_text (textwrap .dedent (source ))
1127
1123
1128
- session = build (paths = tmp_path )
1129
-
1130
- assert session .exit_code == ExitCode .OK
1131
- assert len (session .tasks ) == 4
1132
- assert len (session .tasks [0 ].produces ["return" ]) == 2
1133
- assert len (session .tasks [1 ].depends_on ["paths" ]) == 2
1124
+ result = runner .invoke (cli , [tmp_path .as_posix ()])
1125
+ assert result .exit_code == ExitCode .OK
1126
+ assert "4 Collected tasks" in result .output
1127
+ assert "4 Succeeded" in result .output
1134
1128
assert tmp_path .joinpath ("a-copy.txt" ).exists ()
1135
1129
assert tmp_path .joinpath ("b-copy.txt" ).exists ()
1136
1130
1137
1131
1138
1132
@pytest .mark .end_to_end ()
1139
- def test_delayed_task_generation_with_single_function (tmp_path ):
1133
+ def test_delayed_task_generation_with_single_function (runner , tmp_path ):
1140
1134
source = """
1141
1135
from typing_extensions import Annotated
1142
1136
from pytask import DirectoryNode, task
@@ -1160,17 +1154,15 @@ def task_copy(
1160
1154
"""
1161
1155
tmp_path .joinpath ("task_module.py" ).write_text (textwrap .dedent (source ))
1162
1156
1163
- session = build (paths = tmp_path )
1164
-
1165
- assert session .exit_code == ExitCode .OK
1166
- assert len (session .tasks ) == 3
1167
- assert len (session .tasks [0 ].produces ["return" ]) == 1
1168
- assert len (session .tasks [1 ].depends_on ["paths" ]) == 1
1157
+ result = runner .invoke (cli , [tmp_path .as_posix ()])
1158
+ assert result .exit_code == ExitCode .OK
1159
+ assert "3 Collected tasks" in result .output
1160
+ assert "3 Succeeded" in result .output
1169
1161
assert tmp_path .joinpath ("a-copy.txt" ).exists ()
1170
1162
1171
1163
1172
1164
@pytest .mark .end_to_end ()
1173
- def test_delayed_task_generation_with_task_node (tmp_path ):
1165
+ def test_delayed_task_generation_with_task_node (runner , tmp_path ):
1174
1166
source = """
1175
1167
from typing_extensions import Annotated
1176
1168
from pytask import DirectoryNode, TaskWithoutPath, task, PathNode
@@ -1196,12 +1188,10 @@ def task_depends(
1196
1188
"""
1197
1189
tmp_path .joinpath ("task_module.py" ).write_text (textwrap .dedent (source ))
1198
1190
1199
- session = build (paths = tmp_path )
1200
-
1201
- assert session .exit_code == ExitCode .OK
1202
- assert len (session .tasks ) == 3
1203
- assert len (session .tasks [0 ].produces ["return" ]) == 1
1204
- assert len (session .tasks [1 ].depends_on ["paths" ]) == 1
1191
+ result = runner .invoke (cli , [tmp_path .as_posix ()])
1192
+ assert result .exit_code == ExitCode .OK
1193
+ assert "3 Collected tasks" in result .output
1194
+ assert "3 Succeeded" in result .output
1205
1195
assert tmp_path .joinpath ("a-copy.txt" ).exists ()
1206
1196
1207
1197
0 commit comments