code: def test_get_udf_wrapper():
      fp_config = tdh.create_fp_config(
          mode=FeatureProcessorMode.PYSPARK
      )
      udf_wrapper = Mock(UDFWrapper)
  
      with patch.object(
          UDFWrapperFactory,
          "_get_spark_udf_wrapper",
          return_value=udf_wrapper,
      ) as get_udf_wrapper_method:
          result = UDFWrapperFactory.get_udf_wrapper(fp_config)
  
          assert result == udf_wrapper
          get_udf_wrapper_method.assert_called_with(fp_config)
  
file_path: sagemaker-python-sdk/tests/unit/sagemaker/feature_store/feature_processor/test_factory.py
imports: from __future__ import absolute_import
  import pytest
  import test_data_helpers
  from mock import Mock, patch
  from sagemaker.feature_store.feature_processor._enums import FeatureProcessorMode
  from sagemaker.feature_store.feature_processor._factory import UDFWrapperFactory, ValidatorFactory
  from sagemaker.feature_store.feature_processor._feature_processor_config import FeatureProcessorConfig
  from sagemaker.feature_store.feature_processor._udf_wrapper import UDFWrapper
  from sagemaker.feature_store.feature_processor._validation import FeatureProcessorArgValidator, InputValidator, SparkUDFSignatureValidator, InputOffsetValidator, BaseDataSourceValidator
  from sagemaker.session import Session
label_desc: Use image node_id 2 for calling a global function with example usage: test_get_udf_wrapper() without return types