def test_run(self):
with self.test_session():
gan = mock_gan()
args = hc.Config({"size": "1"})
cli = hg.CLI(gan, args)
cli.run()
self.assertEqual(cli.gan, gan)
def test_train(self):
with self.test_session():
gan = mock_gan()
args = hc.Config({
"size": "1",
"steps": 1,
"method": "train",
"save_every": -1
})
cli = hg.CLI(gan, args)
cli.train()
self.assertEqual(cli.gan, gan)
def test_sample(self):
with self.test_session():
gan = mock_gan()
args = hc.Config({
"size": "1",
"steps": 1,
"method": "train",
"save_every": -1
})
gan.create()
cli = hg.CLI(gan, args)
cli.sample('/tmp/test-sample.png')
self.assertEqual(cli.gan, gan)
def test_adds_supervised_loss(self):
with self.test_session():
gan = mock_gan(y=2)
args = hc.Config({
"size": "1",
"steps": 1,
"method": "train",
"save_every": -1,
"classloss": True
})
cli = hg.CLI(gan, args)
cli.add_supervised_loss()
self.assertEqual(type(cli.gan.loss), MultiComponent)
self.assertEqual(type(cli.gan.loss.components[0]), SupervisedLoss)
def test_new(self):
with self.test_session():
try:
os.remove('test.json')
except Exception:
pass
gan = mock_gan()
args = hc.Config({
"size": "1",
"steps": 1,
"method": "train",
"directory": "test"
})
cli = hg.CLI(gan, args)
cli.new()
self.assertTrue(os.path.isfile('test.json'))
def test_safe_new(self):
with self.test_session():
try:
os.remove('test.json')
except Exception:
pass
gan = mock_gan()
args = hc.Config({
"size": "1",
"steps": 1,
"method": "train",
"directory": "test"
})
cli = hg.CLI(gan, args)
cli.new()
with self.assertRaises(ValidationException):
cli.new()
def test_cli(self):
with self.test_session():
gan = hg.GAN()
args = {}
cli = hg.CLI(gan, args)
self.assertEqual(cli.gan, gan)