30a5671297517875caacbf6bd03ae1de0d94ec67,testing/test_mean_functions.py,TestMeanFuncs,test_mult_output_shape,#TestMeanFuncs#,67

Before Change

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

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

After Change

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

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

Instances