Python script for testing pyspark installation

test_pyspark_dep.py

# Maybe required:
# import os
# os.environ["JAVA_HOME"] = "/usr/lib/jvm/java-8-openjdk-amd64"
# os.environ["SPARK_HOME"] = "/opt/spark"
# Test installation
# import findspark
# findspark.init()

import pyspark
import random

num_samples = 1000000
sc = pyspark.SparkContext(appName="Pi")

def inside(p):     
  x, y = random.random(), random.random()
  return x*x + y*y < 1

count = sc.parallelize(range(0, num_samples)).filter(inside).count()
pi = 4 * count / num_samples
print(pi)
sc.stop()
# 3.14