Source code for denspp.offline.template.run_tests

import unittest
from shutil import rmtree
from denspp.offline.logger import define_logger_testing



[docs]
def load_tests(loader, standard_tests, pattern):
    testcase = unittest.TestLoader()
    suite = unittest.TestSuite()
    suite.addTests(testcase.discover(start_dir='src', pattern='*_test.py'))
    suite.addTest(testcase.discover(start_dir='src_dnn', pattern='*_test.py'))
    suite.addTest(testcase.discover(start_dir='src_pipe', pattern='*_test.py'))
    return suite



if __name__ == '__main__':
    logger = define_logger_testing()
    unittest.main()
    rmtree('temp_test', ignore_errors=True)