Fix up train_{disc,gen}.*_D unit tests
Fix up train_{disc,gen}.*_D unit tests

Some of them were failing because they were trying to distinguish
the same policy from itself. Others were using more steps than
they need to.

HumanCompatibleAI/imitation,HumanCompatibleAI/imitation,humancompatibleai/imitation,humancompatibleai/imitation