d2cdade362349254a6774411df0ded6c6da455d8,tests/samplers/static_batch_sampler_test.py,StaticBatchSamplerTest,test_config,#StaticBatchSamplerTest#,11

Before Change

class StaticBatchSamplerTest(tf.test.TestCase):
    def test_config(self):
        with self.test_session():
            gan = hg.GAN(graph={"x" : tf.constant(1., shape=[1,4,4,1], dtype=tf.float32)})
            gan.create()

            sampler = StaticBatchSampler(gan)
            self.assertEqual(sampler.sample("/tmp/test.png")[0]["image"].shape[-1], 1)

After Change

class StaticBatchSamplerTest(tf.test.TestCase):
    def test_config(self):
        with self.test_session():
            gan = mock_gan()
            gan.create()

            sampler = StaticBatchSampler(gan)
            self.assertEqual(sampler.sample("/tmp/test.png")[0]["image"].shape[-1], 1)

Instances