939c04db549a03f2f4821648f3bb832a2d894d6a,tests/distributions/test_transforms.py,TransformTests,test_discrete_cosine,#TransformTests#,202

Before Change

    def test_discrete_cosine(self):
        // NOTE: Need following since helper function unimplemented
        self._test(lambda input_dim: T.DiscreteCosineTransform())
        self._test(lambda input_dim: T.DiscreteCosineTransform(smooth=0.5))
        self._test(lambda input_dim: T.DiscreteCosineTransform(smooth=1.0))
        self._test(lambda input_dim: T.DiscreteCosineTransform(smooth=2.0))

    def test_haar_transform(self):

After Change

    def test_discrete_cosine(self):
        // NOTE: Need following since helper function unimplemented
        for smooth in [0.0, 0.5, 1.0, 2.0]:
            self._test(lambda input_dim: T.DiscreteCosineTransform(smooth=smooth), autodiff=False)

    def test_haar_transform(self):
        // NOTE: Need following since helper function unimplemented
        for flip in [True, False]:
            self._test(lambda input_dim: T.HaarTransform(flip=flip), autodiff=False)

Instances