use testlib everywhere

1 files changed, 4 insertions, 15 deletions

diff --git a/test/unittest_graph.py b/test/unittest_graph.py
index a68eece..347fa53 100644
--- a/test/unittest_graph.py
+++ b/test/unittest_graph.py
@@ -1,10 +1,9 @@
 # unit tests for the cache module
 
+from logilab.common.testlib import TestCase
 from logilab.common import get_cycles
-import sys
-import unittest
 
-class getCycleTestCase(unittest.TestCase):
+class getCycleTestCase(TestCase):
 
     def test_known0(self):
         self.assertEqual(get_cycles({1:[2], 2:[3], 3:[1]}), [[1, 2, 3]])
@@ -15,18 +14,8 @@ class getCycleTestCase(unittest.TestCase):
     def test_known2(self):
         self.assertEqual(get_cycles({1:[2], 2:[3], 3:[0], 0:[]}), [])
         
-def suite():
-    loader = unittest.TestLoader()
-    testsuite = loader.loadTestsFromModule(sys.modules[__name__])
-    return testsuite
-    
-    
-def Run():
-    testsuite = suite()
-    runner = unittest.TextTestRunner()
-    return runner.run(testsuite)
 
 if __name__ == "__main__":
-    Run()
- 
+    unittest_main()
+