7679bdc57a29f1195e1b11b6b140abd559ae8dea,art/classifiers/wrapper_unittest.py,TestMixinWKerasClassifier,test_layers,#TestMixinWKerasClassifier#,237

Before Change

        self.assertTrue(keras_model._output.name, "output0")

    def test_layers(self):
        self._test_layers(custom_activation=False)
        self._test_layers(custom_activation=True)

    def _test_layers(self, custom_activation=False):
        // Get MNIST

After Change

    def test_layers(self):
        // Get MNIST
        (_, _), (x_test, _), _, _ = load_mnist()
        x_test = x_test[:NB_TEST]

        classifier = ClassifierWrapper(self.model_mnist)
        self.assertEqual(len(classifier.layer_names), 3)

Instances