from pyspark.sql import SparkSession

spark = SparkSession.builder.appName("sg-local-check").getOrCreate()
spark.sparkContext.setLogLevel("ERROR")

total = spark.range(1, 6).selectExpr("sum(id) AS total").first().total
print(f"total={total}")

spark.stop()