def test_create_component(self):
     with self.test_session():
         gan = BaseGAN(inputs=MockInput())
         distribution = gan.create_component(gan.config.distribution)
         self.assertEqual(
             type(distribution),
             hg.distributions.uniform_distribution.UniformDistribution)
Exemple #2
0
 def test_create_component(self):
     with self.test_session():
         gan = BaseGAN(inputs=MockInput())
         encoder = gan.create_component(gan.config.encoder)
         self.assertEqual(type(encoder),
                          hg.encoders.uniform_encoder.UniformEncoder)