tests/runtests.py

   1 #!/usr/bin/env python
   2 # -*- Mode: Python -*-
   3
   4 from __future__ import absolute_import
   5
   6 import os
   7 import sys
   8
   9 import pytest
  10
  11
  12 def main(argv):
  13     if '--help' in argv:
  14         print("Usage: ./runtests.py <testfiles>")
  15         return
  16
  17     mydir = os.path.dirname(os.path.abspath(__file__))
  18
  19     verbosity_args = []
  20
  21     if 'PYGI_TEST_VERBOSE' in os.environ:
  22         verbosity_args += ['--capture=no']
  23
  24     if 'TEST_NAMES' in os.environ:
  25         names = os.environ['TEST_NAMES'].split()
  26     elif 'TEST_FILES' in os.environ:
  27         names = []
  28         for filename in os.environ['TEST_FILES'].split():
  29             names.append(filename[:-3])
  30     elif len(argv) > 1:
  31         names = []
  32         for filename in argv[1:]:
  33             names.append(filename.replace('.py', ''))
  34     else:
  35         return pytest.main([mydir] + verbosity_args)
  36
  37     def unittest_to_pytest_name(name):
  38         parts = name.split(".")
  39         parts[0] = os.path.join(mydir, parts[0] + ".py")
  40         return "::".join(parts)
  41
  42     return pytest.main([unittest_to_pytest_name(n) for n in names] + verbosity_args)
  43
  44
  45 if __name__ == "__main__":
  46     sys.exit(main(sys.argv))