from pyspark.sql import SparkSession

spark = (
    SparkSession.builder
    .master("local[2]")
    .appName("sg-local-check")
    .getOrCreate()
)

result = spark.range(1, 6).selectExpr("sum(id) AS total").collect()[0]["total"]
print(f"local_sum={result}")
print(f"spark_version={spark.version}")
spark.stop()