code: def test_training_input_all_arguments():
      prefix = "pre"
      distribution = "FullyReplicated"
      compression = "Gzip"
      content_type = "text/csv"
      record_wrapping = "RecordIO"
      s3_data_type = "Manifestfile"
      input_mode = "Pipe"
      result = TrainingInput(
          s3_data=prefix,
          distribution=distribution,
          compression=compression,
          input_mode=input_mode,
          content_type=content_type,
          record_wrapping=record_wrapping,
          s3_data_type=s3_data_type,
      )
      expected = {
          "DataSource": {
              "S3DataSource": {
                  "S3DataDistributionType": distribution,
                  "S3DataType": s3_data_type,
                  "S3Uri": prefix,
              }
          },
          "CompressionType": compression,
          "ContentType": content_type,
          "RecordWrapperType": record_wrapping,
          "InputMode": input_mode,
      }
  
      assert result.config == expected
  
file_path: sagemaker-python-sdk/tests/unit/test_inputs.py
imports: from __future__ import absolute_import
  import pytest
  from sagemaker import TrainingInput
  from sagemaker.inputs import FileSystemInput
label_desc: Use image node_id 2 for calling a global function with example usage: test_training_input_all_arguments() without return types