diff options
author | Lukas Geiger <lukas.geiger94@gmail.com> | 2018-07-15 23:36:37 +0200 |
---|---|---|
committer | Lukas Geiger <lukas.geiger94@gmail.com> | 2018-07-15 23:36:37 +0200 |
commit | 006b8faeb79c8b9329bd600390dbda888e9df226 (patch) | |
tree | 4fc4e41632538f59a0c6f60e15d6b298ab2161a6 /tensorflow/contrib/gan | |
parent | e5945c00148186808e337b4946cf0fa6460f6803 (diff) |
[tfgan] Fix assertion in regularization unittest
`self.assertTrue(3.0)` always evaluates to `True`.
This PR fixes assertion so it will correctly check the loss value.
Diffstat (limited to 'tensorflow/contrib/gan')
-rw-r--r-- | tensorflow/contrib/gan/python/train_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/gan/python/train_test.py b/tensorflow/contrib/gan/python/train_test.py index 3ebbe55d05..e905a911cf 100644 --- a/tensorflow/contrib/gan/python/train_test.py +++ b/tensorflow/contrib/gan/python/train_test.py @@ -447,8 +447,8 @@ class GANLossTest(test.TestCase): reg_loss_gen_np = reg_loss.generator_loss.eval() reg_loss_dis_np = reg_loss.discriminator_loss.eval() - self.assertTrue(3.0, reg_loss_gen_np - no_reg_loss_gen_np) - self.assertTrue(3.0, reg_loss_dis_np - no_reg_loss_dis_np) + self.assertEqual(3.0, reg_loss_gen_np - no_reg_loss_gen_np) + self.assertEqual(2.0, reg_loss_dis_np - no_reg_loss_dis_np) def test_regularization_gan(self): self._test_regularization_helper(get_gan_model) |