cis-kuldeep
/
PyBitmessage-2021-04-27
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Improved unit tests loading

This commit is contained in:
Biryuzovye Kleshni 2018-06-23 13:19:59 +00:00
parent 2eeb58a43e
commit 43f7e1e7a1
1 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/workprover/test.py
View File

@ -185,21 +185,22 @@ class TestWorkProver(unittest.TestCase):
self.thread.commandsQueue.put(("shutdown", ))
self.thread.join()
loader = unittest.TestLoader()
suite = unittest.TestSuite([
loader.loadTestsFromTestCase(TestUtils),
loader.loadTestsFromTestCase(TestDumbSolver),
loader.loadTestsFromTestCase(TestForkingSolver),
loader.loadTestsFromTestCase(TestFastSolver),
loader.loadTestsFromTestCase(TestGPUSolver),
loader.loadTestsFromTestCase(TestWorkProver)
])
def load_tests(loader, tests, pattern):
return unittest.TestSuite([
loader.loadTestsFromTestCase(TestUtils),
loader.loadTestsFromTestCase(TestDumbSolver),
loader.loadTestsFromTestCase(TestForkingSolver),
loader.loadTestsFromTestCase(TestFastSolver),
loader.loadTestsFromTestCase(TestGPUSolver),
loader.loadTestsFromTestCase(TestWorkProver)
])
if __name__ == "__main__":
import multiprocessing
multiprocessing.freeze_support()
loader = unittest.TestLoader()
runner = unittest.TextTestRunner()
runner.run(suite)
runner.run(load_tests(loader, [], None))