File: test_package.py

package info (click to toggle)
pytorch 1.13.1%2Bdfsg-4
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 139,252 kB
  • sloc: cpp: 1,100,274; python: 706,454; ansic: 83,052; asm: 7,618; java: 3,273; sh: 2,841; javascript: 612; makefile: 323; xml: 269; ruby: 185; yacc: 144; objc: 68; lex: 44
file content (16 lines) | stat: -rw-r--r-- 509 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
# Owner(s): ["oncall: package/deploy"]

def load_tests(loader, standard_tests, pattern):
    """Load all tests from `test/pacakge/`
    """
    if pattern is None:
        # Use the default pattern if none is specified by the test loader.
        pattern = "test*.py"
    package_tests = loader.discover("package", pattern=pattern)
    standard_tests.addTests(package_tests)
    return standard_tests


if __name__ == "__main__":
    from torch.testing._internal.common_utils import run_tests
    run_tests()