Skip to content

Commit 73cae9e

Browse files
committed
add interruptible checks to models unit test
Signed-off-by: redartera <120470035+redartera@users.noreply.github.com>
1 parent bcbb991 commit 73cae9e

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tests/flytekit/unit/models/test_execution.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -166,6 +166,7 @@ def test_execution_spec(literal_value_pair):
166166
),
167167
raw_output_data_config=_common_models.RawOutputDataConfig(output_location_prefix="raw_output"),
168168
max_parallelism=100,
169+
interruptible=True
169170
)
170171
assert obj.launch_plan.resource_type == _identifier.ResourceType.LAUNCH_PLAN
171172
assert obj.launch_plan.domain == "domain"
@@ -183,6 +184,7 @@ def test_execution_spec(literal_value_pair):
183184
]
184185
assert obj.disable_all is None
185186
assert obj.max_parallelism == 100
187+
assert obj.interruptible == True
186188
assert obj.raw_output_data_config.output_location_prefix == "raw_output"
187189

188190
obj2 = _execution.ExecutionSpec.from_flyte_idl(obj.to_flyte_idl())
@@ -203,6 +205,7 @@ def test_execution_spec(literal_value_pair):
203205
]
204206
assert obj2.disable_all is None
205207
assert obj2.max_parallelism == 100
208+
assert obj2.interruptible == True
206209
assert obj2.raw_output_data_config.output_location_prefix == "raw_output"
207210

208211
obj = _execution.ExecutionSpec(
@@ -220,6 +223,7 @@ def test_execution_spec(literal_value_pair):
220223
assert obj.metadata.principal == "tester"
221224
assert obj.notifications is None
222225
assert obj.disable_all is True
226+
assert obj.interruptible is None
223227

224228
obj2 = _execution.ExecutionSpec.from_flyte_idl(obj.to_flyte_idl())
225229
assert obj == obj2
@@ -233,6 +237,7 @@ def test_execution_spec(literal_value_pair):
233237
assert obj2.metadata.principal == "tester"
234238
assert obj2.notifications is None
235239
assert obj2.disable_all is True
240+
assert obj2.interruptible is None
236241

237242

238243
def test_workflow_execution_data_response():

0 commit comments

Comments
 (0)