diff options
author | Ezio Melotti <ezio.melotti@gmail.com> | 2013-03-01 11:23:28 +0200 |
---|---|---|
committer | Ezio Melotti <ezio.melotti@gmail.com> | 2013-03-01 11:23:28 +0200 |
commit | f79493bf9ebb191f06083fdcd0bce4626641b4ac (patch) | |
tree | 036c52baa9fd0bc239a8ffc67c2f271f1c6a4c8d /Lib/test/test_dbm.py | |
parent | 520378f81d30105d421e7b3a53c30fb0346be269 (diff) | |
download | cpython-git-f79493bf9ebb191f06083fdcd0bce4626641b4ac.tar.gz |
#17082: test_dbm* now work with unittest test discovery. Patch by Zachary Ware.
Diffstat (limited to 'Lib/test/test_dbm.py')
-rw-r--r-- | Lib/test/test_dbm.py | 20 |
1 files changed, 10 insertions, 10 deletions
diff --git a/Lib/test/test_dbm.py b/Lib/test/test_dbm.py index 02df7e3cab..752e0b77a6 100644 --- a/Lib/test/test_dbm.py +++ b/Lib/test/test_dbm.py @@ -34,7 +34,7 @@ def delete_files(): test.support.unlink(f) -class AnyDBMTestCase(unittest.TestCase): +class AnyDBMTestCase: _dict = {'0': b'', 'a': b'Python:', 'b': b'Programming', @@ -119,10 +119,6 @@ class AnyDBMTestCase(unittest.TestCase): class WhichDBTestCase(unittest.TestCase): - # Actual test methods are added to namespace after class definition. - def __init__(self, *args): - unittest.TestCase.__init__(self, *args) - def test_whichdb(self): for module in dbm_iterator(): # Check whether whichdb correctly guesses module name @@ -169,12 +165,16 @@ class WhichDBTestCase(unittest.TestCase): self.d.close() -def test_main(): - classes = [WhichDBTestCase] +def load_tests(loader, tests, pattern): + classes = [] for mod in dbm_iterator(): - classes.append(type("TestCase-" + mod.__name__, (AnyDBMTestCase,), + classes.append(type("TestCase-" + mod.__name__, + (AnyDBMTestCase, unittest.TestCase), {'module': mod})) - test.support.run_unittest(*classes) + suites = [unittest.makeSuite(c) for c in classes] + + tests.addTests(suites) + return tests if __name__ == "__main__": - test_main() + unittest.main() |