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)
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)