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)