vscode-python/pythonFiles/tests/__main__.py at master · thy09/vscode-python · GitHub

45 lines (32 loc) · 1.18 KB

# Licensed under the MIT License.

import argparse

import sys

import pytest

from . import DEBUG_ADAPTER_ROOT, IPYTHON_ROOT, SRC_ROOT, TEST_ROOT, TESTING_TOOLS_ROOT

def parse_args():

parser = argparse.ArgumentParser()

# To mark a test as functional: (decorator) @pytest.mark.functional

parser.add_argument(

"--functional", dest="markers", action="append_const", const="functional"

)

parser.add_argument(

"--no-functional", dest="markers", action="append_const", const="not functional"

)

args, remainder = parser.parse_known_args()

ns = vars(args)

return ns, remainder

def main(pytestargs, markers=None):

sys.path.insert(1, IPYTHON_ROOT)

sys.path.insert(1, TESTING_TOOLS_ROOT)

sys.path.insert(1, DEBUG_ADAPTER_ROOT)

pytestargs = ["--rootdir", SRC_ROOT, TEST_ROOT] + pytestargs

for marker in reversed(markers or ()):

pytestargs.insert(0, marker)

pytestargs.insert(0, "-m")

ec = pytest.main(pytestargs)

return ec

if __name__ == "__main__":

mainkwargs, pytestargs = parse_args()

ec = main(pytestargs, **mainkwargs)

sys.exit(ec)

Expand file tree