Blame view

3rdparty/opencv-4.5.4/modules/python/test/test.py 1.82 KB
f4334277   Hu Chunming   提交3rdparty
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
  #!/usr/bin/env python
  '''
  Location of tests:
  - <opencv_src>/modules/python/test
  - <opencv_src>/modules/<module>/misc/python/test/
  '''
  
  from __future__ import print_function
  
  import sys
  sys.dont_write_bytecode = True  # Don't generate .pyc files / __pycache__ directories
  
  import os
  import unittest
  
  # Python 3 moved urlopen to urllib.requests
  try:
      from urllib.request import urlopen
  except ImportError:
      from urllib import urlopen
  
  from tests_common import NewOpenCVTests
  
  
  basedir = os.path.abspath(os.path.dirname(__file__))
  
  def load_tests(loader, tests, pattern):
      cwd = os.getcwd()
      config_file = 'opencv_python_tests.cfg'
      locations = [cwd, basedir]
      if os.path.exists(config_file):
          with open(config_file, 'r') as f:
              locations += [str(s).strip() for s in f.readlines()]
      else:
          print('WARNING: OpenCV tests config file ({}) is missing, running subset of tests'.format(config_file))
  
      tests_pattern = os.environ.get('OPENCV_PYTEST_FILTER', 'test_*') + '.py'
      if tests_pattern != 'test_*.py':
          print('Tests filter: {}'.format(tests_pattern))
  
      processed = set()
      for l in locations:
          if not os.path.isabs(l):
              l = os.path.normpath(os.path.join(cwd, l))
          if l in processed:
              continue
          processed.add(l)
          print('Discovering python tests from: {}'.format(l))
          sys_path_modify = l not in sys.path
          if sys_path_modify:
              sys.path.append(l)  # Hack python loader
          discovered_tests = loader.discover(l, pattern=tests_pattern, top_level_dir=l)
          print('    found {} tests'.format(discovered_tests.countTestCases()))
          tests.addTests(loader.discover(l, pattern=tests_pattern))
          if sys_path_modify:
              sys.path.remove(l)
      return tests
  
  if __name__ == '__main__':
      NewOpenCVTests.bootstrap()