[Pix2Pix Zero] Fix slow tests (#2391)
* fix: slow tests.
* retrieving the slices.
* fix: assertion.
* debugging.
* debugging.
* debugging
* debugging.
* debugging
* debugging.
* debugging.
* debugging
* debugging
* change debugging.
* change debugging.
* fix: tests for pix2pix zero.
