Generalize the replay_obs_test a bit with better debugging tools to make it easier to deflake.
Generalize the replay_obs_test a bit with better debugging tools to make it easier to deflake.

PiperOrigin-RevId: 187484720

deepmind/pysc2