Added __call__ for PySpark; factored out tests