def METHOD_NAME(self):
    @dsl.container_component
    def container_comp_with_artifacts(
            dataset: dsl.Input[dsl.Dataset],
            num_epochs: int,  # also as an input
            model: dsl.Output[dsl.Model],
            model_config_path: dsl.OutputPath(str),
    ):
        return dsl.ContainerSpec(
            image='gcr.io/my-image',
            command=['sh', 'run.sh'],
            args=[
                '--dataset_location',
                dataset.path,
                '--epochs',
                num_epochs,
                '--model_path',
                model.uri,
                '--model_config_path',
                model_config_path,
            ])
    self.assertIsInstance(container_comp_with_artifacts,
                          container_component_class.ContainerComponent)