use CleverHansTest

bjtuxck · Aug 31, 2017 · e0036e2 · e0036e2
1 parent 01fa82c
commit e0036e2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 15 deletions.
diff --git a/tests_tf/test_attacks.py b/tests_tf/test_attacks.py
@@ -5,8 +5,8 @@
 
 import unittest
 import numpy as np
-import time
 
+from cleverhans.devtools.checks import CleverHansTest
 from cleverhans.attacks import Attack
 from cleverhans.attacks import FastGradientMethod
 from cleverhans.attacks import BasicIterativeMethod
@@ -15,19 +15,6 @@
 from cleverhans.attacks import CarliniWagnerL2
 
 
-class CleverHansTest(unittest.TestCase):
-    def setUp(self):
-        self.test_start = time.time()
-        # seed the randomness
-        np.random.seed(1234)
-
-    def tearDown(self):
-        print(self.id(), "took", time.time() - self.test_start, "seconds")
-
-    def assertClose(self, x, y):
-        # self.assertTrue(np.allclose(x, y)) doesn't give a useful message
-        # on failure
-        assert np.allclose(x, y), (x, y)
 
 
 class TestAttackClassInitArguments(CleverHansTest):

diff --git a/tests_tf/test_mnist_tutorial_tf.py b/tests_tf/test_mnist_tutorial_tf.py
@@ -1,8 +1,10 @@
 import unittest
 import numpy as np
 
+from cleverhans.devtools.checks import CleverHansTest
 
-class TestMNISTTutorialTF(unittest.TestCase):
+
+class TestMNISTTutorialTF(CleverHansTest):
     def test_mnist_tutorial_tf(self):
 
         np.random.seed(42)