From 006b8faeb79c8b9329bd600390dbda888e9df226 Mon Sep 17 00:00:00 2001 From: Lukas Geiger Date: Sun, 15 Jul 2018 23:36:37 +0200 Subject: [PATCH] [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. --- tensorflow/contrib/gan/python/train_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/contrib/gan/python/train_test.py b/tensorflow/contrib/gan/python/train_test.py index 3ebbe55d059..e905a911cf1 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)