run_tests.py 606 B

123456789101112131415161718192021
  1. import sys
  2. import pytest
  3. from opendevin import config
  4. if __name__ == '__main__':
  5. """Main entry point of the script.
  6. This script runs pytest with specific arguments and configuration.
  7. Usage:
  8. python script_name.py [--OPENAI_API_KEY=<api_key>] [--model=<model_name>]
  9. """
  10. args = ['-v', 'evaluation/regression/cases']
  11. for arg in sys.argv[1:]:
  12. if arg.startswith('--OPENAI_API_KEY='):
  13. config.config['OPENAI_API_KEY'] = arg.split('=')[1]
  14. elif arg.startswith('--model='):
  15. args.append(f'-o model={arg.split('=')[1]}')
  16. pytest.main(args)