Write tests for tensor_based_metric and token_based_metric.

[kushalarora]

https://github.com/kushalarora/lmpl/blob/647ff2fb4de09a39720a25db633b74859b5efa1f/lmpl/modules/decoders/base_rollin_rollout_decoder.py#L588-L588