Clean out module-scope imports: .tests.test_weakset

transaction/tests/test_weakset.py

@@ -11,15 +11,12 @@
 # FOR A PARTICULAR PURPOSE
 #
 ##############################################################################
-
 import unittest
-from transaction.weakset import WeakSet
 
-class Dummy:
-    pass
 
 class WeakSetTests(unittest.TestCase):
     def test_contains(self):
+        from transaction.weakset import WeakSet
         w = WeakSet()
         dummy = Dummy()
         w.add(dummy)
@@ -29,6 +26,7 @@ class WeakSetTests(unittest.TestCase):
 
     def test_len(self):
         import gc
+        from transaction.weakset import WeakSet
         w = WeakSet()
         d1 = Dummy()
         d2 = Dummy()
@@ -40,6 +38,7 @@ class WeakSetTests(unittest.TestCase):
         self.assertEqual(len(w), 1)
 
     def test_remove(self):
+        from transaction.weakset import WeakSet
         w = WeakSet()
         dummy = Dummy()
         w.add(dummy)
@@ -49,6 +48,7 @@ class WeakSetTests(unittest.TestCase):
 
     def test_as_weakref_list(self):
         import gc
+        from transaction.weakset import WeakSet
         w = WeakSet()
         dummy = Dummy()
         dummy2 = Dummy()
@@ -64,6 +64,7 @@ class WeakSetTests(unittest.TestCase):
         self.assertEqual(set(L), set([dummy, dummy2]))
 
     def test_map(self):
+        from transaction.weakset import WeakSet
         w = WeakSet()
         dummy = Dummy()
         dummy2 = Dummy()
@@ -78,9 +79,9 @@ class WeakSetTests(unittest.TestCase):
             self.assertEqual(thing.poked, 1)
         
 
-def test_suite():
-    return unittest.makeSuite(WeakSetTests)
+class Dummy:
+    pass
         
-if __name__ == '__main__':
-    unittest.main()
 
+def test_suite():
+    return unittest.makeSuite(WeakSetTests)