Add test checking if the optimizer variables are written into checkpoint file.
Add test checking if the optimizer variables are written into checkpoint file.

PiperOrigin-RevId: 351806492
Change-Id: I0cdbfcc61c7b6bee30abb875bed7bfe08a4866a1

tensorflow/agents,tensorflow/agents