How to do sanity test for PyTorchJob in Kubeflow properly?

Question

I need to run sanity test to make sure that a PyTorchJob can be created and run till completion in Kubeflow. The job is supposed to be simple and minimal.

Here is the sam