Adding a util based on DocTest to test docstrings and extending docstring for `tft.apply_buckets`.
Adding a util based on DocTest to test docstrings and extending docstring for `tft.apply_buckets`.

PiperOrigin-RevId: 326454451

tensorflow/transform