Add tfds.testing.MockPolicy to public API.
Add tfds.testing.MockPolicy to public API.

PiperOrigin-RevId: 343040369

tensorflow/datasets,tensorflow/datasets,tensorflow/datasets,tensorflow/datasets,tensorflow/datasets