Is your feature request related to a problem? Please describe.
#2993 rewrites code generation for Kubeflow Pipelines That PR includes automated tests for common scenarios, but some scenarios are not covered. We need to add tests for those scenarios.