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.commandsQueue.put(("shutdown", ))
self.thread.join() self.thread.join()
loader = unittest.TestLoader() def load_tests(loader, tests, pattern):
return unittest.TestSuite([
suite = unittest.TestSuite([ loader.loadTestsFromTestCase(TestUtils),
loader.loadTestsFromTestCase(TestUtils), loader.loadTestsFromTestCase(TestDumbSolver),
loader.loadTestsFromTestCase(TestDumbSolver), loader.loadTestsFromTestCase(TestForkingSolver),
loader.loadTestsFromTestCase(TestForkingSolver), loader.loadTestsFromTestCase(TestFastSolver),
loader.loadTestsFromTestCase(TestFastSolver), loader.loadTestsFromTestCase(TestGPUSolver),
loader.loadTestsFromTestCase(TestGPUSolver), loader.loadTestsFromTestCase(TestWorkProver)
loader.loadTestsFromTestCase(TestWorkProver) ])
])
if __name__ == "__main__": if __name__ == "__main__":
import multiprocessing import multiprocessing
multiprocessing.freeze_support() multiprocessing.freeze_support()
loader = unittest.TestLoader()
runner = unittest.TextTestRunner() runner = unittest.TextTestRunner()
runner.run(suite)
runner.run(load_tests(loader, [], None))