pep8 in test_attacks
@@ -14,6 +14,7 @@ from cleverhans.attacks import VirtualAdversarialMethod
from cleverhans.attacks import SaliencyMapMethod
from cleverhans.attacks import CarliniWagnerL2
+
class CleverHansTest(unittest.TestCase):
def setUp(self):
self.test_start = time.time()
@@ -28,6 +29,7 @@ class CleverHansTest(unittest.TestCase):
# on failure
assert np.allclose(x, y), (x, y)
+
class TestAttackClassInitArguments(CleverHansTest):
def test_model(self):
@@ -84,8 +86,6 @@ class TestParseParams(CleverHansTest):
self.assertTrue(test_attack.parse_params({}))
-
-
class TestVirtualAdversarialMethod(CleverHansTest):
def setUp(self):
super(TestVirtualAdversarialMethod, self).setUp()
