code: def test_modify_node():
      modifier = training_params.TrainPrefixRemover()
  
      for params in _parameter_combinations():
          node = ast_call("Estimator({})".format(params))
          modifier.modify_node(node)
  
          expected = "Estimator({})".format(params).replace(
              "train_", ""
          )
          assert expected == pasta.dump(node)
  
file_path: sagemaker-python-sdk/tests/unit/sagemaker/cli/compatibility/v2/modifiers/test_training_params.py
imports: from __future__ import absolute_import
  import itertools
  import pasta
  from sagemaker.cli.compatibility.v2.modifiers import training_params
  from tests.unit.sagemaker.cli.compatibility.v2.modifiers.ast_converter import ast_call
label_desc: Use image node_id 5 for calling a global function with example usage: test_modify_node() without return types