gnattest/testsuite/drivers/python_script.py

import os
import sys

from drivers.base_driver import BaseDriver


class PythonScriptDriver(BaseDriver):
    """
    Driver to run a Python script.

    Interface:

    * put a "test.py" script in the test directory;
    * put a "test.out" text file in the test directory.

    This driver will run the Python script. Its output is then checked against
    the expected output (test.out file). This mechanism is the most flexible
    way to write a testcase, but also the more verbose one and the most complex
    one. Use this driver when no other one fits.
    """

    def run(self):
        env = dict(os.environ)
        old_path = env.get('PYTHONPATH', '')
        if old_path:
            new_path = '{}{}{}'.format(
                self.env.root_dir, os.path.pathsep, old_path)
        else:
            new_path = self.env.root_dir
        env['PYTHONPATH'] = new_path

        self.shell([sys.executable, 'test.py'], env=env)