Skip to content

Commit 96e8b9d

Browse files
feriferi
feri
authored and
feri
committed
add new template_fields to unit tests
1 parent bb1d09c commit 96e8b9d

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

providers/google/tests/unit/google/cloud/operators/test_vertex_ai.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1216,6 +1216,10 @@ def test_templating(self, create_task_instance_of_operator, session):
12161216
region="{{ 'region' }}",
12171217
project_id="{{ 'project_id' }}",
12181218
impersonation_chain="{{ 'impersonation-chain' }}",
1219+
template_path="{{ 'template-path' }}",
1220+
pipeline_root="{{ 'pipeline-root' }}",
1221+
service_account="{{ 'service-account' }}",
1222+
parameter_values={"{{ 'parameter-key' }}": "{{ 'parameter-value' }}"},
12191223
# Other parameters
12201224
dag_id="test_template_body_templating_dag",
12211225
task_id="test_template_body_templating_task",
@@ -1229,6 +1233,10 @@ def test_templating(self, create_task_instance_of_operator, session):
12291233
assert task.region == "region"
12301234
assert task.project_id == "project_id"
12311235
assert task.impersonation_chain == "impersonation-chain"
1236+
assert task.template_path == "template-path"
1237+
assert task.pipeline_root == "pipeline-root"
1238+
assert task.service_account == "service-account"
1239+
assert task.parameter_values == {"parameter-key": "parameter-value"}
12321240
assert task.training_pipeline_id == "training-pipeline-id"
12331241
assert task.custom_job_id == "custom_job_id"
12341242

0 commit comments

Comments
 (0)