code: def setup_module():
      _setup_ray_on_spark_envs()
  
imports: import sys
  import pytest
  import os
  import time
  import functools
  from abc import ABC
  from pyspark.sql import SparkSession
  from ray.tests.spark.test_basic import RayOnSparkCPUClusterTestBase, _setup_ray_cluster, _setup_ray_on_spark_envs, _RAY_ON_SPARK_WORKER_PHYSICAL_MEMORY_BYTES, _RAY_ON_SPARK_WORKER_SHARED_MEMORY_BYTES
  from ray.util.spark.utils import _calc_mem_per_ray_worker_node
  from ray._private.test_utils import wait_for_condition
  import ray
label_desc: Use image node_id 1 for calling a global function with example usage: setup_module() without return types