code: def spark_session(mock_data_frame):
      return Mock(
          SparkSession,
          read=Mock(
              csv=Mock(return_value=Mock()),
              parquet=Mock(return_value=mock_data_frame),
              conf=Mock(set=Mock()),
          ),
          table=Mock(return_value=mock_data_frame),
      )
  
imports: from __future__ import absolute_import
  import pytest
  import test_data_helpers
  from mock import Mock, patch, call
  from pyspark.sql import SparkSession, DataFrame
  from sagemaker.feature_store.feature_processor._data_source import CSVDataSource, FeatureGroupDataSource, ParquetDataSource, IcebergTableDataSource
  from sagemaker.feature_store.feature_processor._input_loader import SparkDataFrameInputLoader
  from sagemaker.feature_store.feature_processor._spark_factory import SparkSessionFactory
  from sagemaker.feature_store.feature_processor._env import EnvironmentHelper
  from sagemaker.session import Session
label_desc: Use image node_id 3 for calling a global function with example usage: spark_session(mock_data_frame) and returns: Mock