c95d470a391216cfe96e717840b2cef8d053f7b1,art/defences/variance_minimization_unittest.py,TestTotalVarMin,test_three_channels,#TestTotalVarMin#,31

Before Change

    def test_three_channels(self):
        (train_features, _), (_, _) = cifar10.load_data()
        x = train_features[:2] / 255.0
        preprocess = TotalVarMin()
        preprocessed_x = preprocess(x)
        self.assertTrue((preprocessed_x.shape == x.shape))

After Change

        self.assertFalse((preprocessed_x == x).all())

    def test_three_channels(self):
        x = np.random.rand(2, 32, 32, 3)
        preprocess = TotalVarMin()
        preprocessed_x = preprocess(x)
        self.assertTrue((preprocessed_x.shape == x.shape))
        self.assertTrue((preprocessed_x <= 1.0).all())

Instances