code: def create_pyspark_dataframe_with_mocked_map_data() -> (
      PySparkDataFrame
  ):
      """Returns PySpark DataFrame with mocked map data."""
      spark = SparkSession.builder.appName(
          "snowflake.com"
      ).getOrCreate()
      map_schema = StructType(
          [
              StructField("lat", FloatType(), True),
              StructField("lon", FloatType(), True),
          ]
      )
      return spark.createDataFrame(data=MAP_DATA, schema=map_schema)
  
file_path: streamlit/lib/tests/streamlit/pyspark_mocks.py
imports: from __future__ import annotations
  import numpy
  import pandas
  from pyspark.sql.dataframe import DataFrame
  from pyspark.sql.session import SparkSession
  from pyspark.sql.types import FloatType, IntegerType, StringType, StructField, StructType
label_desc: Use image node_id 2 for calling a global function with example usage: create_pyspark_dataframe_with_mocked_map_data() and returns: spark