update tests

Signed-off-by: Kevin Su <[email protected]>
flyteorg · Mar 21, 2024 · 167527a · 167527a
1 parent ab82e36
commit 167527a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/plugins/flytekit-spark/tests/test_remote_register.py b/plugins/flytekit-spark/tests/test_remote_register.py
@@ -44,3 +44,7 @@ def my_python_task(a: str) -> int:
 
     # Check if the serialized python task has no mainApplicaitonFile field set by default.
     assert serialized_spec.template.custom is None
+
+    remote.register_task(my_python_task, version="v1")
+    serialized_spec = mock_client.create_task.call_args.kwargs["task_spec"]
+    assert serialized_spec.template.custom is None