feature: Add tests for RegisterModel with repack output

Author: tuliocasagrande

tests/unit/sagemaker/workflow/test_step_collections.py

@@ -802,6 +802,32 @@ def test_register_model_with_model_repack_with_pipeline_model(
             raise Exception("A step exists in the collection of an invalid type.")
 
 
+def test_register_model_with_model_repack_with_repack_output_path(model):
+    repack_output_path = "s3://{BUCKET}/repack_output"
+    register_model = RegisterModel(
+        name="RegisterModelStep",
+        model=model,
+        content_types=["content_type"],
+        response_types=["response_type"],
+        inference_instances=["inference_instance"],
+        transform_instances=["transform_instance"],
+        model_package_group_name="mpg",
+        approval_status="Approved",
+        description="description",
+        depends_on=["TestStep"],
+        tags=[{"Key": "myKey", "Value": "myValue"}],
+        repack_output_path=repack_output_path,
+    )
+
+    request_dicts = register_model.request_dicts()
+
+    for request_dict in request_dicts:
+        if request_dict["Type"] == "Training":
+            arguments = request_dict["Arguments"]
+            assert arguments["DebugHookConfig"]["S3OutputPath"] == repack_output_path
+            assert arguments["OutputDataConfig"]["S3OutputPath"] == repack_output_path
+
+
 def test_estimator_transformer(estimator):
     model_data = f"s3://{BUCKET}/model.tar.gz"
     model_inputs = CreateModelInput(