9332d1d8fee62ad6d2d0e7f066917082e3b534e3,tests/losses/softmax_loss_test.py,SoftmaxLossTest,test_create,#SoftmaxLossTest#,17

Before Change

    def test_create(self):
        with self.test_session():
            graph = {}
            graph["d_real"] = tf.constant(0, shape=[2,2])
            graph["d_fake"] = tf.constant(0, shape=[2,2])
            print("gcraph", graph)
            loss = SoftmaxLoss(hg.GAN(graph=graph), loss_config)
            d_loss, g_loss = loss.create()
            d_shape = loss.ops.shape(d_loss)
            g_shape = loss.ops.shape(g_loss)

After Change

    def test_create(self):
        with self.test_session():
            graph = mock_graph()

            gan = hg.GAN(graph=graph)
            gan.create()
            loss = SoftmaxLoss(gan, loss_config)
            d_loss, g_loss = loss.create()
            d_shape = loss.ops.shape(d_loss)
            g_shape = loss.ops.shape(g_loss)
            self.assertEqual(sum(d_shape), 1)

Instances