30a5671297517875caacbf6bd03ae1de0d94ec67,testing/test_mean_functions.py,TestMeanFuncs,test_basic_output_shape,#TestMeanFuncs#,49

Before Change

    def test_basic_output_shape(self):
        with self.test_context() as sess:
            for mf in self.mfs1:
                with mf.tf_mode():
                    Y = sess.run(
                        mf(self.X),
                        feed_dict={self.x: mf.get_free_state(), self.X: self.X_data})
                self.assertTrue(Y.shape in [(self.N, self.output_dim), (self.N, 1)])

    def test_add_output_shape(self):
        with self.test_context() as sess:

After Change

    def test_basic_output_shape(self):
        with self.test_context() as sess:
            X = tf.placeholder(settings.tf_float, shape=[self.N, self.input_dim])
            X_data = np.random.randn(self.N, self.input_dim).astype(settings.np_float)
            for mf in self.mfs1():
                mf.compile()
                Y = sess.run(mf(X), feed_dict={X: X_data})
                self.assertTrue(Y.shape in [(self.N, self.output_dim), (self.N, 1)])

    def test_add_output_shape(self):
        with self.test_context() as sess:

Instances