[TF Agents] Clean up some DQN Agent unit test code.
[TF Agents] Clean up some DQN Agent unit test code.

PiperOrigin-RevId: 262058937
Change-Id: I8239751eac002388ef13d96d2717af4e16557083

tensorflow/agents,tensorflow/agents