Update checkpointing API docs in Python/Java.
Clean up setup code in PySpark checkpointing tests