mirror of
https://github.com/kforney/pentest-distro-builder.git
synced 2024-10-06 20:48:51 -06:00
27 lines
582 B
Python
27 lines
582 B
Python
|
def main():
|
||
|
import os
|
||
|
import sys
|
||
|
from ptvsd.visualstudio_py_debugger import DONT_DEBUG, DEBUG_ENTRYPOINTS, get_code
|
||
|
|
||
|
sys.path[0] = os.getcwd()
|
||
|
os.chdir(sys.argv[1])
|
||
|
testFx = sys.argv[2]
|
||
|
args = sys.argv[3:]
|
||
|
|
||
|
DONT_DEBUG.append(os.path.normcase(__file__))
|
||
|
DEBUG_ENTRYPOINTS.add(get_code(main))
|
||
|
|
||
|
try:
|
||
|
if testFx == 'pytest':
|
||
|
import pytest
|
||
|
pytest.main(args)
|
||
|
else:
|
||
|
import nose
|
||
|
nose.run(argv=args)
|
||
|
sys.exit()
|
||
|
finally:
|
||
|
pass
|
||
|
|
||
|
if __name__ == '__main__':
|
||
|
main()
|