code: def test_training_input_all_defaults(caplog):
      prefix = "pre"
      actual = TrainingInput(s3_data=prefix)
      expected = {
          "DataSource": {
              "S3DataSource": {
                  "S3DataDistributionType": "FullyReplicated",
                  "S3DataType": "S3Prefix",
                  "S3Uri": prefix,
              }
          }
      }
      assert actual.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 1 for calling a global function with example usage: test_training_input_all_defaults(caplog) without return types