TF: add beam search tests
# What does this PR do?

Adds `generate()` beam search tests, in advance of the beam search refactor.
_The documentation is not available anymore as the PR was closed or merged._